1. 程式人生 > 實用技巧 >pytorch環境搭建(Windows10、CPU版)

pytorch環境搭建(Windows10、CPU版)

系統:windows10(x64)

之所以選擇裝Anaconda來安裝torch等環境,主要是由於conda可以自動安裝依賴庫,免於pip安裝時出現的各種錯誤,節約時間。

0.解除安裝之前的python

1.安裝Anaconda(手動新增環境變數)

2.修改conda映象源配置

3.建立conda的python虛擬環境

4.安裝pytorch

5.配置pycharm編譯環境

(步驟0不贅述)

1.安裝Anaconda

  參考:https://www.jianshu.com/p/d3a5ec1d9a08

  去官網下載對應版本(x64)的anaconda,下載過程比較慢。安裝時記住不要勾選Add path to,要後面自己手動配置:

  安裝好之後將如下三個環境變數加入系統變數中(根據自己的安裝路徑變化):

  D:\Anaconda

  D:\Anaconda\Scripts

  D:\Anaconda\Library\bin

  之後在cmd中確認conda已經成功安裝:

2.修改conda映象源配置

  此步開始為後續安裝各種庫做準備。由於預設源的下載速度通常緩慢,因此更換為清華源(或科大源等):

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/
conda config --add channels http://
mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/win-64/ conda config --set show_channel_urls yes conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/

P.S.以防在後續建立環境和下載安裝pytorch的時候出現HTTP 000 CONNECTION FAILED for url“<https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/pytorch-1.6.0-py3.8_cpu_0.tar.bz2>”類似的錯誤,需要將源地址中的https(如果有的話)改成http,且要在源地址後加上/win-64/

  換回預設源指令:

conda config --remove-key channels

  參考:https://blog.csdn.net/weixin_38359037/article/details/106910996

     https://www.cnblogs.com/tianlang25/p/12433025.html

3.建立conda的python虛擬環境

  根據自己的python版本,建立虛擬環境(其中灰色部分為所建立環境的名稱)

#建立環境
conda create -n pytorch_cpu python=3.8
#啟用環境
activate pytorch_cpu
#退出啟用環境
deactivate

4.安裝pytorch

  在pytorch官網上(www.pytorch.org)找到自己需要的版本,按照所給指令複製到cmd中安裝

(numpy在安裝pytorch的過程中已經安裝了)

#檢視已安裝的庫
conda list

5.配置pycharm編譯環境

  File>>Settings>>Project:xxxxx>>Project Interpreter>>Conda Environment>>Existing environment>>D:\Anaconda\envs\pytorch_cpu/python.exe(根據自己的安裝目錄修改,envs即為之前建立的環境所在資料夾)

#測試torch是否安裝成功
import torch
x=torch.rand(5,3)
print(x)