PyTorch學習筆記2--PyTorch環境的搭建

環境選擇

Anaconda + CUDA9.2 + PyCharm

• Anaconda3-5.2.0-Windows-x86_64.exe;
• cuda_9.2.148.1_windows.exe和cuda_9.2.148.1_windows.exe；

注意安裝Anaconda后如果環境變量PATH中沒有Anaconda3.5目錄,Anaconda3.5\Library\bin目錄以及Anaconda\Scripts目錄，要添加到Path環境變量。

依次安裝Anaconda3安裝程序和CUDA9.2的驅動程序。

1 如果已經配置過tensorflow，那么可以直接安裝Pytorch了。

2 強烈建議離線安裝的方式，省時省心！

3 在base環境下安裝pytorch包即可，不要新建envs

在線安裝PyTorch

1. 配置conda的軟件源：在C:\Users\YourID\.condarc文件內容改為如下內容后保存退出：

   channels:
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
     - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
     - defaults
   show_channel_urls: true
   ssl_verify: true
   
   

2. 進入Anaconda Prompt命令行程序，檢查Python版本 python --version

3. 使用下列命令下載安裝torch和torchvision：

   #默認 使用 cuda10.1
   conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
   
   #cuda 9.2
   conda install pytorch torchvision cudatoolkit=9.2 -c pytorch -c defaults -c numba/label/dev
   
   #cpu版本
   conda install pytorch torchvision cpuonly -c pytorch
   

離線安裝PyTorch和torchvision：

如果上面的下載過程很慢：

• 進入PyTorch官網，依次選擇你電腦的配置
• 然后根據配置，去頁面https://download.pytorch.org/whl/torch_stable.html選擇文件：找到CUDA版本為9.2，Python版本為3.6，平台為Win64的1.4.0版本的Torchtorch-1.4.0%2Bcu92-cp36-cp36m-win_amd64.whl文件,將其下載下來。
• 安裝: 進入Anaconda Prompt命令行程序，pip install torch-1.4.0+cu92-cp36-cp36m-win_amd64.whl即可。
• 安裝完成后會提示msgpack找不到，需要再額外安裝一個，去https://pypi.tuna.tsinghua.edu.cn/simple下載一個msgpack-0.6.2-cp36-cp36m-win_amd64.whl，運行pip install msgpack-0.6.2-cp36-cp36m-win_amd64.whl安裝好即可。
• 去https://pypi.org/project/torchvision/0.5.0/#files下載后pip install torchvision-0.5.0-cp36-cp36m-win_amd64.whl進行安裝。

下載壓縮包安裝文件后運行conda install --use-local ****.bz2也可

驗證環境

輸入python，進入PYTHON環境

import torch
torch.__version__
# 得到結果'1.3.0'
print(torch.cuda.is_available())
# 得到結果True

使用pycharm開發

如果你不喜歡Spyder或jupyter，可以使用pycharm進行開發。pycharm專業版對於學校教職工和學生是免費的。

1. Pycham下載: https://www.jetbrains.com/pycharm/download/#section=windows
2. 安裝: 這里就不必細說，直接next就OK,中間有一步勾選.py文件關聯。
3. windows ：打開hosts文件，文件的最后一行添加：0.0.0.0 account.jetbrains.com
4. 去jetbrains官網申請注冊一個學生賬戶:https://www.jetbrains.com/shop/eform/students
   5.重啟Pycharm，選擇 登陸賬戶。

點擊首頁的Configure，選Setting，點Project Interpreter設置解釋器，右側選擇...\Anaconda3\python.exe，可以看到如下圖所示，能夠識別到torch和torchvision等Packages.

注意，Pycharm第一次啟動后會自動更新其內部的程序，請等待完成后再開始編寫你的代碼。使用View>Scientigic Mode，Variable中的變量可以右鍵View as Array.
然后即可開心玩耍啦~