win10在Anaconda上安裝Pytorch

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環境:

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章