Anaconda創建pytorch環境


前提:找好自己python對應的anaconda版本,不然很麻煩

 

我們首先使用 conda 指令創建一個屋子,叫做 pytorch。

指令如下:

conda create -n pytorch python=3.6

conda 是指調用 conda 包,create 是創建的意思,-n 是指后面的名字是屋子的名字, pytorch是屋子的名字(可以更改成自己喜歡的),python=3.6 是指創建的屋子,是 python3.6 版本。

如果報錯:

打開清華大學鏡像源,網站如下:
https://mirrors.tuna.tsinghua.edu.cn/help/anaconda/
在這里插入圖片描述
將框中所有代碼復制到.condarc文件中,該文件位於
C:\Users\用戶名
位置,如果沒有,則手動創建並進行粘貼。

之后便可愉快的創建新環境啦!

之后,彈出提示,輸入 y,即可安裝。

安裝成功后,輸入以下指令:

conda info --envs 

即可看到 conda 環境中,有新建的 pytorch 環境,右邊的 * 號表示,當前你處於哪個環境。

接下來,我們要在 pytorch 環境中,安裝 PyTorch,(有點繞),使用如下指令,進入 pytorch 環境。

conda activate pytorch 

你可以看到左邊的 base 變成了 pytorch,代表成功進入 pytorch 環境。

安裝 PyTorch 准備

終於到重頭戲了,安裝 PyTorch了。激動的打開官網,下拉到下面的頁面。

(不得不說,PyTorch 在這方面做的真的好,不需要再人工安裝 CUDA、cuDNN 之類的,全部都給你解決了。真的良心~以前安裝真的繁瑣。(吐槽一句,以前超麻煩的)

PyTorch Build 選擇 Stable;選擇系統;Package,Windows下推薦 conda,Linux 下推薦 pip;Python版本按照Anaconda的版本選擇,我這里選擇3.6,CUDA 推薦9.2。如果沒有顯卡的話,選擇 None。我的選擇如圖所示:

復制下面的代碼,之后,在開始菜單中,打開 Anaconda Prompt,查看最左邊括號中是 base 還是 pytorch。

如果是 base,使用 conda activate pytorch 進入 pytorch 環境中。之后粘貼即可。

果斷輸入y,之后就是漫長的等待。或者去休息吧,慢慢等着。看看劇啥的,等着它慢慢下吧。如果你的並不慢,恭喜恭喜。

驗證是否安裝成功

(1)在命令行左邊為 pytorch 環境中,輸入 python

(2)之后,輸入 import torch,如果沒有報錯,意味着 PyTorch 已經順利安裝了

(3)接下來,輸入 torch.cuda.is_available,如果是 True,意味着你可以使用 GPU,如果是 False,意味着只能使用CPU。


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM