3060顯卡tensorflow 安裝


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

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM