1、3060+window10+cuda11.1+cudnn8.0.4+conda+tensorflow2.4.1 

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')]