Win10系統(tǒng)1050ti顯卡安裝cuda11.5和cudnn8.4,TensorFlow2.6.0
本文只是針對1050ti顯卡在2022年4月9日的安裝方法,采用python3.7,cuda11.5,cudnn8.4,vs2022。本文最終只支持GPU True tf.Tensor(6.0, shape=(), dtype=float32)的結(jié)果,Anaconda和pycharm的安裝以及具體的細(xì)節(jié)強(qiáng)烈建議先看這位up的視頻https://www.bilibili.com/video/BV1Lv41177BW。
?

1.確定安裝版本
在桌面用win+R輸入cmd進(jìn)入控制臺,輸入nvidia-smi

或者在桌面右鍵進(jìn)入英偉達(dá)控制面板

看到圖中支持的版本是cuda11.5,接著在TensorFlow官網(wǎng) https://tensorflow.google.cn/install/source_windows查看對應(yīng)的cudnn版本。

目前還沒有找到11.5,所以我選擇了python3.7、cudnn在官網(wǎng)找的8.4版本,vs之前用的時候已經(jīng)裝了2022版也沒有卸載。
2. 安裝cuda11.5和cudnn8.4
在官網(wǎng)https://developer.nvidia.com/cuda-toolkit-archive選擇cuda11.5

在cudnn官網(wǎng)NVIDIA cuDNN | NVIDIA Developerhttps://developer.nvidia.com/cudnn選擇8.4下載


至此,針對1050ti的兩個文件為:

雙擊cuda的exe先解壓,然后安裝,然后把第二個壓縮包的三個文件放到cuda的安裝目錄里面粘貼替換。

在控制臺輸入nvcc -V查看是否安裝成功

?
3.創(chuàng)建conda虛擬環(huán)境(具體細(xì)節(jié)請參考視頻)
在桌面用win+R輸入cmd進(jìn)入控制臺;
控制臺輸入conda create -n tf26 python==3.7,輸入y,創(chuàng)建完成;
輸入activate進(jìn)入base環(huán)境;
輸入conda activate tf26進(jìn)入虛擬環(huán)境,環(huán)境名稱就是tf26,可以任意;
輸入
pip install tensorflow_gpu==2.6.0 -i https://pypi.douban.com/simple --trusted-host pypi.douban
或者
pip install tensorflow_gpu==2.6.0 -i https://pypi.douban.com/simple
即可進(jìn)行安裝TensorFlow2.6。文件大概340Mb.
?
測試是否安裝成功可以參見https://blog.csdn.net/qq_35456045/article/details/104666083
4.在pycharm中添加python解釋器
在pycharm中依次點(diǎn)擊:文件-設(shè)置-python解釋器-顯示所有-添加之前創(chuàng)建的虛擬環(huán)境tf26-確定(如下圖所示),需要等pycharm掃描這個解釋器,等右下角進(jìn)度條跑完,強(qiáng)烈建議先重啟pycharm。


然后輸入代碼。
import tensorflow as tf
?
#查看tensorflow版本
print(tf.__version__)
?
print('GPU', tf.test.is_gpu_available())
?
a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)
?

如圖,顯示gpu false,計算結(jié)果為2+4=6.
?如果遇到cudart64_110.dll 文件不存在,下載后放到c\windows\system32下面https://www.dll-files.com/cudart64_101.dll.html
?
?
?
主要參考以下資料:
?
【包教包會】Anaconda+Cuda+Cudnn+TensorFlow+Pytorch+Pycharm+Win10深度學(xué)習(xí)環(huán)境配置與卸載https://www.bilibili.com/video/BV1Lv41177BW
Install Tensorflow GPU on Windows https://www.youtube.com/watch?v=EmZZsy7Ym-4
error: Ignore above cudart dlerror if you do not have a GPU set up on your machine. https://blog.csdn.net/qq_43355165/article/details/121879462
TensorFlow官網(wǎng) https://tensorflow.google.cn/install/source_windows
安裝cuda和cudnn https://blog.csdn.net/linxinloningg/article/details/122522975
TensorFlow2.x GPU版安裝與CUDA版本選擇指南 https://www.csdn.net/tags/MtjaAg5sMDk4ODQtYmxvZwO0O0OO0O0O.html
cudart64_110.dll not found windows解決方法 https://blog.csdn.net/weixin_42474121/article/details/115708919
Win10+1050Ti配置tensorflow-gpu教程 (解決1050ti配置cuda失敗的問題)https://blog.csdn.net/mch2869253130/article/details/88889616
win10 1050Ti 筆記本配置 TensorFlow-gpu 過程(多圖超詳細(xì))https://blog.csdn.net/weixin_43626465/article/details/115706195
AttributeError: module 'tensorflow' has no attribute 'Session'錯誤解決 https://blog.csdn.net/sinat_36502563/article/details/102302392
?最后,本文只是針對1050ti顯卡在2022年4月9日的安裝,采用python3.7,cuda11.5,cudnn8.4,vs2022。本文最終只支持GPU True tf.Tensor(6.0, shape=(), dtype=float32)的結(jié)果,不過后來聽說測試結(jié)果為GPU false 其實也不影響運(yùn)行,其他代碼還沒有測試。