安装之前将原来的python以及环境变量路径删除,Anaconda自带特定版本的python
下载Anaconda最新版本 https://www.anaconda.com/distribution/#download-section
前面默认安装,但注意需要将anaconda加入环境变量,勾选红色框中的选项,等待安装完毕
开始菜单打开Anaconda Prompt,新建一个名叫TF2.1的环境
conda create -n TF2.1 python=3.7
进入环境
conda activate TF2.1
接下来安装NVIDIA SDK10.1版本,但是安装之前可以选用清华源以便快速安装
# 输入下面三行命令即可
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes
# 有时候国内镜像源无法连接,需要恢复原来的源:
conda config --remove-key channels
那么开始安装NVIDIA SDK吧
conda install cudatoolkit=10.1
接着安装NVIDIA 深度学习软件包
conda install cudnn=7.6
如果上面两个安装步骤出现问题,请检查设备是否包含NVIDIA GPU,若不支持,则跳过这两步,直接安装TF
pip install tensorflow==2.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
终于安装完毕了,测试一下
python
import tensorflow as tf
tf.__version__
若打印出TF版本号,则安装成功;但若出现:
ImportError: DLL load failed: 找不到指定的模块。
那么可以尝试降低版本为TF2.0
输入一下命令,将自动卸载TF2.1并安装TF2.0
# cpu版本
pip install tensorflow==2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
# gpu版本
pip install tensorflow-gpu==2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
如果遇到找不到cudart64_100.dll,可以将包含该动态链接库的路径添加到系统环境变量
例如:C:\Users\KIM\anaconda3\envs\TF2.1\Library\bin
祝君好运!