1. 程式人生 > 實用技巧 >關於anaconda 安裝 tensorflow gpu問題

關於anaconda 安裝 tensorflow gpu問題

一般情況下我會採用ancanda直接進行安裝;

原因一:無需為cuda和cudnn另外去下載和配置,ancanda可以在虛擬環境裡面為您通過命令的方式直接安裝好。

原因二:目前的pytorch和百度的飛槳框架都可以通過conda命令一鍵安裝好環境,包括(cuda和cudnn)非常方便。

tensorflow需要注意的地方:安裝tensorflow的時候猶豫沒有conda安裝命令,貌似官方也沒有推薦使用conda安裝,這裡就會給安裝tensorflow帶來一定的疑惑。

其實實際上tensorflow是可以使用conda進行安裝的,我們可以先建立一個虛擬環境:

conda create -n tf36 python=3.6

然後啟用該環境:

conda activate tf36

接下來安裝cuda和cudnn,還是還是使用conda命令安裝(因為安裝超級方便,無需我們自己手動去下載,而且必須先安裝好cuda和cudnn再去安裝tensorflow,否則會失敗)

conda install cudatoolkit=10.0
conda install cudnn=7.6.5

然後,我們再通過pip命令的方式進行安裝tensorflow:(requirements.txt,這個是因為我的專案是已經配置好環境的,直接通過requirements.txt進行安裝,如果您不懂requirements.txt請百度)

pip install -r requirements.txt

這樣的話,我們就安裝好了tensorflow的環境,而且是在虛擬環境裡面安裝的cuda不會影響我們其他的環境。非常完美!!