cuda 和cudnn下载百度网盘链接
链接:https://pan.baidu.com/s/1WHMjE70veSo8R6Hvjfl6_w
提取码:81wp
2、先安装cuda11.1,再复制cudnn8.0.4的目录到cuda1.1相应的目录下。(注,后期TensorFlow安装时可能遇见一个BUG)
3、conda安装
4、tensorflow 安装
1)先创建conda虚拟环境(可以不创建)
conda create -n tensorflow python=3.7
activate tensorflow
conda deactivate
2)安装tensorflow-gpu
pip install tensorflow-gpu==2.4.1 -i https://pypi.tuna.tsinghua.edu.cn/simple
3) 报错:Could not load dynamic library 'cusolver64_10.dll'; dlerror: cusolver64_10.dll not found
运行tf.config.list_physical_devices('GPU'),时报上面错误
复制C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin\cusolver64_11.dll
为cusolver64_10.dll即可。
4)成功
>>> tf.config.list_physical_devices('GPU')
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]