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