文章目录
- 1. 安装CUDA
- 2.cudnn安装
- 3.tensorflow-gpu安装
- 4.再把缺失的文件cudart64_100.dll放在bin文件夹下
- 5.切换版本
1. 安装CUDA
网址:
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
1.
2. https://blog.csdn.net/sinat_23619409/article/details/84202651
2.cudnn安装
NVIDIA cuDNN是用于深度神经网络的GPU加速库
安装方法也在上边的网址里
3.tensorflow-gpu安装
使用清华镜像:
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
4.再把缺失的文件cudart64_100.dll放在bin文件夹下
5.切换版本
现在有了tensorflow 1.15.0的cpu版本,还有gpu2.0.0的版本,想要切换通过激活:
activate tensorflow/activate tensorflow-gpu
mmp 我失败了,只保留了gpu版本的