小白的pytorch安裝記錄

小白pytorch安裝

• 完全的代碼小白pytorch安裝記錄
• • Anaconda3安裝
  • Pytorch安裝
  • Pycharm安裝

完全的代碼小白pytorch安裝記錄

我是真的沒什麼經驗，這裡面的知識可能不夠權威，僅供參考。

Anaconda3安裝

官網下載下傳安裝即可

1. 打開anaconda的prompt；
2. 檢視自己的電腦是否支援gpu版本，我的電腦支援，并且是NVIDIA960M；
3. conda create -n pytorch python=3.5；（我使用的Python3.5）
4. conda activate pytorch；

Pytorch安裝

1. 打開pytorch官網選擇适合自己的版本，這裡往往會很慢，是以采用清華鏡像

2. 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

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

4. conda install pytorch torchvision cudatoolkit=10.1

   上述依次複制粘貼即可

   這裡需要自己的GPU驅動在400+，否則可能出現錯誤，我的電腦就是需要去NVIDIA更新自己的驅動，更新完就OK了。

   我這裡一開始倔強用官網方法，一下午都沒搞定，用清華鏡像幾分鐘搞定。

Pycharm安裝

1.官網下載下傳安裝，我選擇免費的社群版。至此，經測試我的安裝完成且成功。