anaconda安裝gpu版本的tensorflow2.0.0教程

曉碼bigdata發表於2020-11-16

anaconda安裝gpu版本的tensorflow教程

1、安裝anaconda
2、建立虛擬環境
3、安裝tensorflow及相關包
4、測試是否執行正常

1、安裝anaconda
直接百度,教程很簡單

2、建立虛擬環境
開啟anaconda的anaconda prompt,輸入以下conda命令:
conda create -n tensorflow2.0gpu python=3.6
跳出詢問是否下載後,輸入y下載

3、安裝tensorflow及相關包
(1)檢視自己的顯示卡是否支援gpu加速,一般算力要大於3.5
(2)開啟命令列cmd,輸入nvidia-smi,NVIDIA驅動程式需大於410.x版本
在這裡插入圖片描述
(3)開啟anaconda prompt,輸入activate tensorflow2.0gpu啟用建立的虛擬環境,分別輸入conda install cudatoolkit;和conda install cudnn=7.6.0下載好這兩個包
在這裡插入圖片描述
(4)直接下載gpu版本的tensorflow
pip install tensorflow-gpu==2.0.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

4、測試是否執行正常
在自己的編輯器中,配置好環境,
import tensorflow as tf
tf.version #檢視版本
tf.test.is_gpu_available() #檢視gpu是否可用

通過以上測試基本ok,但是後續執行中可能會遇到cudnn無法加速的問題,一般原因是顯示卡已經爆了,具體再百度。
有問題歡迎諮詢qq1269143766,同時執行各種深度學習代做。

相關文章