最近买了台本做训练用,因为一直没GPU,今天晚上配置一下TensorFlow环境,配置如下:
官方下载链接:https://visualstudio.microsoft.com/downloads/?rr=https%3A%2F%2Fwww.baidu.com%2Flink%3Furl%3D7607zWVCfrNBww7i-jC9Idd12bUvZTWfWfQ74KDiDRaVRT9ySKsskOPUcKEByZdTqpRaFQXiNPwQ5-JwXD8nm_%26wd%3D%26eqid%3Dbf2cd077000268f2000000035bf57bdb
百度网盘链接:https://pan.baidu.com/s/1ESJnnxpO2sRey0eE1Am4xA
工作负载选择如下图所示:
CUDA版本选择9.0,对应的CUDNN版本为7.4,下面直接给出百度网盘的下载地址。默认安装即可
百度网盘下载:https://pan.baidu.com/s/1Ak21bYpIPd-H-sjLQ498Dg
百度网盘下载地址: https://pan.baidu.com/s/1WZkEjTHgTAEKlPNCWDu5GQ
下载解压后(里边有个三个文件)拷贝到cuda9.0文件下 ,如下图所示:
官网下载地址(速度比较慢): https://www.anaconda.com/download/
百度网盘下载地址:https://pan.baidu.com/s/1qDe2mEoFt9cjjQS0B0poDA
安装需注意将Anaconda添加到PATH环境变量中,勾选第一个复选框。
以管理员身份运行cmd,然后配置清华镜像,输入指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.6
激活并进入环境,使后续指令在激活的环境中生效,输入指令:
conda activate tensorflow-gpu
升级pip到最新版,防止稍后的安装时,出现错误(笔者初次在安装tensorflow时,没有更新pip到最新版,导致下载到一半出现错误),输入指令:
python -m pip install --upgrade pip
安装TensorFlow GPU版本,输入命令:
pip install tensorflow-gpu