Pytorch官網信息
打開 PyTorch 的官網https://pytorch.org/ , 往下拉找到下如圖所示的界面。你需要根據自己機器的情況,回答下面幾個問題,以便獲得適合你的 PyTorch 安裝命令:
• PyTorch Build 版本,你需要安裝穩定版還是預覽版
• Your OS,你的操作系統是什麼
• Package,你準備用什麼包管理器
• Language,你準備用什麼語言以及什麼版本
• CUDA,你的 CUDA的版本是什麼
CUDA的安裝詳細經過和報錯處理可以參考:https://blog.csdn.net/wumenglu1018/article/details/88170519
下圖就是我當前的環境,即win10+Anaconda+python3.6+cuda9.0
所以安裝就直接按照官網提供的命令即可。
創建Pytorch虛擬環境
在安裝 PyTorch 之前最好先創建一個 pytorch 的虛擬環境。之所以創建虛擬環境是因爲 Python 爲不同的項目需求創建不同的虛擬環境非常常見。在實際項目開發中,我們通常會根據自己的需求去下載各種相應的框架庫,但是可能每個項目使用的框架庫並不一樣,或使用框架的版本不一樣,這樣需要我們根據需求不斷的更新或卸載相應的庫,管理起來相當麻煩。所以通過創建虛擬環境,相當於爲不同的項目創建一塊獨立的空間,在這個空間裏,你安裝任何庫和框架都是獨立的,不會影響到外部環境。
總之就是想象爲 PyTorch 單獨創建一個空間,這個空間與其它空間互不影響。例如安裝TensorFlow的時候也是創建了一個虛擬環境對吧。
打開Anaconda Prompt,輸入命令:
conda create --name pytorch python=3.6
這裏的pytorch是虛擬環境的名稱,可以自己隨意取名。
然後進入pytorch虛擬環境:
activate pytorch
安裝pytorch
輸入上述官網提供的安裝命令:
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
安裝報錯,pytorch包無法下載安裝:
CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/win-64/pytorch-1.0.1-py3.6_cuda90_cudnn7_1.tar.bz2>
Elapsed: -
An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.
因爲官方給出的安裝命令中最後的-c pytorch,是默認指定anaconda官網上的安裝源,可能由於牆的問題,無法正常或者迅速下載。
查看C:\Users\你的用戶名下\.condarc文件,如果有-default那一行,就給刪掉,並指定清華開源鏡像的地址。
我的.condarc文件是沒有問題的,所以去掉官網執行命令的-c pytorch,再次執行,發現仍然報錯,仍然沒有使用指定的清華的安裝源。
直接在命令行中指定,將anconda的安裝源修改爲清華的鏡像源:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
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/pkgs/free/ conda
config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/ conda
config --set show_channel_urls yes
然後執行pytorch的安裝命令:
conda install pytorch torchvision cudatoolkit=9.0
結果終於成功!!!!!!
驗證pytorch的安裝
進入python環境:
python
然後,繼續輸入:
import torch
import torchvision
不報錯就表明安裝成功
查看pytorch環境: