讓人頭疼的tensorflow2.0安裝,確定nVidia的驅動版本直接跳過通過conda安裝虛擬環境,捎帶pytorch

聲明:轉載的原文地址是?https://www.cnblogs.com/xxxms/p/11826388.html?,感謝作者分享自己的經驗,非常感謝

發布這篇文章是因為這么好的文章百度起來是如此的困難,請大家多支持發布這種節省生命的文章的作者,作為一個程序呆子,醉心于技術,真心希望能夠少花點時間在一些根本不具備參考價值的過程上,謝謝大家支持。

windows10下安裝tensorflow2.0-GPU和Cupy(不用搞CUDA+cudnn)

0.前言

今年暑假買了個1660ti的游戲本學python,后來發現跑一些數據量比較大的代碼和深度學習的時候太慢了,遂想裝一下GPU版本,看了網上的資料搞了好幾天,又是CUDA又是cudnn的,網速慢不說,裝完還各種報錯,最后發現隨著anaconda和tensorflow的更新,現在利用anadonda只要一行命令就能完成。(裝好虛擬環境后,直接看4.2)

  

1. 安裝anaconda

  這部分沒啥難度,只要注意兩點:

1.去官網下載https://www.anaconda.com/distribution/,而不要去清華的鏡像站(已經很久沒有更新了),官網下比清華慢一點,但相信我這是值得的。

    2.安裝的時候建議直接把兩個勾都選上(網上的其他教程都建議不勾選然后手動添加PATH,但我發現直接勾上沒問題,之后加PATH反而有各種問題出現)



    #.如果之前電腦里已經有安裝python,建議先卸載再裝anaconda,這樣能避免很多問題


    做完這一步你就擁有了一個預裝了很多包的base環境,包括pandas和numpy,足夠初學者做很多事情了。


2. 換清華源

  雖然上面說anaconda的安裝包不要去清華上下,但是用鏡像站來下載和更新python包還是可以的(能快上不少)。打開CMD,復制打入就行

            conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

            conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge?

            conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/

            conda config --set show_channel_urls yes


3.創建一個另外的虛擬環境

  起先看網上教程的時候一直不明白這一步有什么用,處理了很多報錯以后發現這是為了避免依賴包版本沖突。例如你在官網上下載的anaconda,自帶的base環境是python3.8,而tensorflow是建議裝python3.6,它要求的各種依賴包也是python3.6時代的,直接在base環境下安裝,需要將本來是最新的軟件包降級,可能會影響其他包,最新的功能就用不到了。

  所以創建虛擬環境可以完全避免依賴包沖突,需要使用tensorflow的時候切換過來就行。

  3.1 打開CMD(如果你把anaconda裝在了C盤,建議使用管理員運行,可以事先避免權限報錯),鍵入:

            conda create -n env_gpu python=3.6?

 (env_gpu是你給虛擬環境起的名字,依個人喜好就行,網上大多叫tensorflow或者env_tf2,我這邊叫env_gpu是因為我用這個環境來跑需要gpu加速的任務:tensorflow-gpu和cupy)



?  按y回車,anaconda就會安裝一些必須的包(換源之后是很快的)

4. 在虛擬環境下安裝tensorflow-GPU(用于機器學習)和cupy(numpy的GPU加速版本)

    4.1 首先激活虛擬環境:

         conda activate env_gpu

         (整個第4步的cmd都別關掉,用同一個,假如手滑關了,就要再激活一次虛擬環境)


可以看到前面有個括號表明所在的環境


4.2.安裝tensorflow-gpu和cupy

            conda install tensorflow-gpu=2.0.0

            conda install cupy

            (一行一行來)




可以看到anaconda會自動安裝匹配版本的cuda和cudnn,不用去英偉達官網下載了,也避免了操心版本問題。

4.3 安裝spyder, jupyter notebook

          因為我們在第三步創建了一個獨立的虛擬環境,和原來base環境里的spyder和jupyter notebook是不通用的,所以在這個環境下手動安裝一個就行:

          conda install spyder

          conda install jupyter


          安裝完以后可以在開始菜單搜索spyder,(anaconda)結尾的是base環境下的spyder, (env_gpu)結尾的是這個環境下的spyder。


5.測試一下吧?

    5.1 打開spyder(env_gpu)

    5.2 鍵入一些測試的code


    5.3 cupy也導入測試下



都沒有報錯,大功告成!

conda虛擬環境操作

1. conda env list 或 conda info -e 查看當前存在哪些虛擬環境

2. conda update conda 檢查更新當前conda

3. conda update --all 更新本地已安裝的包

4. conda create -n your_env_namepython=X.X(2.7、3.6等) anaconda 命令創建python版本為X.X、名字為your_env_name的虛擬環境。your_env_name文件可以在Anaconda安裝目錄envs文件下找到。

5. Windows: activate your_env_name(虛擬環境名稱) 激活虛擬環境

6. conda install -n your_env_name [package] 安裝package到your_env_name中

7.linux: source deactivate?????????? Windows: deactivate???? 關閉虛擬環境

8. conda remove -n your_env_name(虛擬環境名稱) --all 刪除虛擬環境 如果出錯試試 conda env remove -n your_env_name

9. conda remove --name your_env_name package_name? 刪除環境中的某個包


conda info --envs 顯示虛擬環境列表


使用清華鏡像快速安裝pytorch

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

conda config --set show_channel_urls yes

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

conda install pytorch torchvision cudatoolkit=XX.0 (版本號從https://pytorch.org/上,查找對應版本)

下面就可以快速安裝了。

若果做GNN推薦是用DGL框架,新手友好。

我的是conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

-c pytorch必須刪除,否則清華鏡像無法使用,下載失敗

conda create -n env_pytorch python=3.6

conda install pytorch torchvision cudatoolkit=10.1

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

不管是pytorch還是tensorflow2.0我都建議安裝spyder和jupyter notebook,集成的環境還是比較容易使用的.

conda install spyder

conda install jupyter

匯總下,避免混淆

conda create -n env_pytorch python=3.6

conda install pytorch torchvision cudatoolkit=10.1

conda install spyder

conda install jupyter

接下來可以進入python看看pytorch是否安裝成功,輸入下述命令,若無報錯則安裝成功:

import torch

import? torchvision

下面是一些gpu相關的命令:torch.cuda.is_available()

cuda是否可用;

torch.cuda.device_count()

返回gpu數量;

torch.cuda.get_device_name(0)

返回gpu名字,設備索引默認從0開始;

torch.cuda.current_device()

返回當前設備索引;


如果想要安裝pytorch+pandas需要的步驟

conda create -n env_pytorch python=3.6

conda install pandas

conda install pytorch torchvision cudatoolkit=10.1

conda install spyder

conda install jupyter

測試import pandas as pd只要不出提示信息表示正常


PS:

進入虛擬環境

conda activate env_gpu

conda activate env_pytorch

退出虛擬環境

conda deactivate

IDE應用虛擬環境

你可以在 VSCode 中?Ctrl+Shift+P?,鍵入?select interpreter?來選擇你想要的 python 解釋器而無視激活的環境,然后Ctrl+F5運行,code的F5調試經常出問題


你同樣可以在pycharm中按Ctrl+Alt+S中選擇切換虛擬環境(第一次創建虛擬環境必須添加虛擬環境),然后Shift+F9調試,Shift+F10運行



python的話,建議使用pycharm,比較好用一些,相比pycharm,code有些地方做的不是很好,比如項目

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。
  • 序言:七十年代末,一起剝皮案震驚了整個濱河市,隨后出現的幾起案子,更是在濱河造成了極大的恐慌,老刑警劉巖,帶你破解...
    沈念sama閱讀 228,786評論 6 534
  • 序言:濱河連續發生了三起死亡事件,死亡現場離奇詭異,居然都是意外死亡,警方通過查閱死者的電腦和手機,發現死者居然都...
    沈念sama閱讀 98,656評論 3 419
  • 文/潘曉璐 我一進店門,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人,你說我怎么就攤上這事。” “怎么了?”我有些...
    開封第一講書人閱讀 176,697評論 0 379
  • 文/不壞的土叔 我叫張陵,是天一觀的道長。 經常有香客問我,道長,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 63,098評論 1 314
  • 正文 為了忘掉前任,我火速辦了婚禮,結果婚禮上,老公的妹妹穿的比我還像新娘。我一直安慰自己,他們只是感情好,可當我...
    茶點故事閱讀 71,855評論 6 410
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著,像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上,一...
    開封第一講書人閱讀 55,254評論 1 324
  • 那天,我揣著相機與錄音,去河邊找鬼。 笑死,一個胖子當著我的面吹牛,可吹牛的內容都是我干的。 我是一名探鬼主播,決...
    沈念sama閱讀 43,322評論 3 442
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場噩夢啊……” “哼!你這毒婦竟也來了?” 一聲冷哼從身側響起,我...
    開封第一講書人閱讀 42,473評論 0 289
  • 序言:老撾萬榮一對情侶失蹤,失蹤者是張志新(化名)和其女友劉穎,沒想到半個月后,有當地人在樹林里發現了一具尸體,經...
    沈念sama閱讀 49,014評論 1 335
  • 正文 獨居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,833評論 3 355
  • 正文 我和宋清朗相戀三年,在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 43,016評論 1 371
  • 序言:一個原本活蹦亂跳的男人離奇死亡,死狀恐怖,靈堂內的尸體忽然破棺而出,到底是詐尸還是另有隱情,我是刑警寧澤,帶...
    沈念sama閱讀 38,568評論 5 362
  • 正文 年R本政府宣布,位于F島的核電站,受9級特大地震影響,放射性物質發生泄漏。R本人自食惡果不足惜,卻給世界環境...
    茶點故事閱讀 44,273評論 3 347
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧,春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,680評論 0 26
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至,卻和暖如春,著一層夾襖步出監牢的瞬間,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,946評論 1 288
  • 我被黑心中介騙來泰國打工, 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留,地道東北人。 一個月前我還...
    沈念sama閱讀 51,730評論 3 393
  • 正文 我出身青樓,卻偏偏與公主長得像,于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子,可洞房花燭夜當晚...
    茶點故事閱讀 48,006評論 2 374

推薦閱讀更多精彩內容