還在系統(tǒng)中安裝CUDA?Anaconda虛擬環(huán)境中安裝CUDA配合Tensorflow-gpu-多版本

先導(dǎo)篇

如果你看到這篇文章,那么說明你已經(jīng)節(jié)省了大半天時(shí)間會(huì)花費(fèi)在developer.nvidia.com網(wǎng)站上了!好用請點(diǎn)贊,收藏評(píng)論??!有問題請留言~

目前裝CUDA有兩種模式:
1.windows=>CUDA
2.windows=>anaconda=>CUDA (教程是基于這種模式,windows\Linux通用,cuda安裝有anaconda管理,不直接安裝于系統(tǒng)中!?。?br> 查閱網(wǎng)上的Tensorflow-gpu的安裝教程,模式一裝cuda時(shí)會(huì)讓人看到頭禿,因?yàn)樗麄兌际亲屇闳ス倬W(wǎng)下載cuda和cudnn,基于Windows系統(tǒng),然后各種添加環(huán)境變量,各種版本必須兼容,會(huì)把你搞得身心俱疲。如果你使用了anaconda,那么其實(shí)cuda套件安裝過程十分簡單?!緵]安裝的話,想簡單就上WEST先生的車吧??!mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,清華鏡像源走一波,下對(duì)應(yīng)版本的exe安裝即可,如下圖anaconda3-2020.11版本】

image.png

參考你所需要的Tensorflow版本,安裝對(duì)應(yīng)版本的cuda,這點(diǎn)很重要!?。?!【10.1、10.2、11等版本適用于Tensorflow2.0以上】
image.png

假設(shè)當(dāng)前你已經(jīng)完成了虛擬環(huán)境的創(chuàng)建(參考我的前一篇文章),如tf1

conda create -n  tf1    python=3.7

activate激活虛擬環(huán)境,然后安裝tensorflow和tensorflow-gpu命令如下:

1.安裝tensorflow

conda install  tensorflow-gpu  #默認(rèn)最新版本 當(dāng)前為cuda 11.0
or
conda install  tensorflow-gpu==1.14.0

2.安裝cudatoolkit

conda install -c anaconda cudatoolkit  #-c代表--channel。 它用于指定搜索包的頻道  默認(rèn)最新版本
or
conda install cudatoolkit=10.0 # 若需指定版本

3.安裝cudnn

conda install -c anaconda cudnn  #若上步指定了版本,這里將做自動(dòng)匹配~
conda install  cudnn  = 7.3.1

4.加入環(huán)境變量

設(shè)置path中的虛擬環(huán)境tf1的路徑 anaconda\envs\tf1\Library\bin


image.png

5.驗(yàn)證可用性

可用jupyter notebook腳本運(yùn)行

import tensorflow as tf
tf.test.is_gpu_available()

返回true即可

6.異常處理

流程1:

新建環(huán)境裝python, 嘗試先安裝對(duì)應(yīng)的cudatoolkit、cudnn(這個(gè)可以conda search cudnn,進(jìn)行小版本選擇,針對(duì)2060我吃過虧,7.6.5大版本下,7.3.1小版本卻不同,低版本才兼容),再裝適合的tensorflow。

流程2:

已知所需tensorflow的版本,查找對(duì)應(yīng)的python版本,新建環(huán)境裝py,再裝對(duì)應(yīng)版本的tensorflow、cuda、cudnn。

這個(gè)安裝天時(shí)地利人和,必須各版本適配。非常建議!!網(wǎng)上查找自己顯卡成功過的適配模式!??!下面分享一個(gè)自己的環(huán)境。愿大家都能成功?。?!

nvidia 2060: python3.7 cuda10.0 cudnn7.3.1

--Mr.west記
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容