TensorFlow-gpu環境配置

1、安裝anaconda

2、新建TensorFlow-gpu環境

3、安裝cuda

(1)查看gpu對應cuda版本:桌面右鍵->NVIDIA控制面板->幫助->系統信息->組件->查看cuda信息(圖1);

(2)下載cuda並安裝:cuda下載地址;(目錄:C:\Program Files\NVIDIA GPU Computing Toolkit)

(3)下載cudnn:cudnn下載地址(需要賬號密碼登錄,qq郵箱、賬號annatu1206、密碼含字母數字特殊字符);

           下載完成後不需要安裝,解壓後替換掉cuda9.0根目錄下的文件就行

4、進入TensorFlow-gpu環境,安裝tensorflow-gpu

(1)升級pip工具防止安裝到一半報錯:python -m pip install --upgrade pip

(2)安裝tensorflow-gpu:pip install --ignore-installed --upgrade tensorflow-gpu

 

遇到的問題:

1、安裝tensorflow-gpu有點問題,import時報錯

原因:安裝的cuda10.1

解決方案:卸載cuda10.1,需保留驅動、系統軟件、vs相關軟件;然後重新安裝cuda10.0

圖1:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章