Windows 10配置TensorFlow-gpu 2.2.0
Windows 10环境下配置TensorFlow-gpu 2.2.0(Nvidia 10/20系显卡适用),已在台式机(GTX 1070Ti)和笔记本(RTX 2070 MAX-Q)上验证。
- 安装Visual Studio 2015及两个运行库vc_redist.x64.exe和vc_redist.x86.exe;
- 安装CUDA 10.1和cuDNN 7.6.5;
- 安装Anaconda3 (Python 3.7版本的);
- 安装/升级依赖的Python包:
python -m pip install --upgrade pip -i https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install --upgrade setuptools -i https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install --upgrade werkzeug -i https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install typed-ast==1.4.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install --upgrade wrapt -i https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install ipykernel -i https://pypi.tuna.tsinghua.edu.cn/simple
python -m pip install tensorflow-gpu==2.2.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
// 可选opencv-python,一般需要用到opencv
python -m pip install opencv-python==3.4.13.47 -i https://pypi.tuna.tsinghua.edu.cn/simple
清华源:https://pypi.tuna.tsinghua.edu.cn/simple
豆瓣源:https://pypi.douban.com/simple
阿里源:https://mirrors.aliyun.com/pypi/simple (貌似更稳定一些)