設備環境:
- windows10
- Anaconda5.2
- python 3.6
更換Anaconda國內源-參考: Anaconda安裝雜記
1. 新建Pytorch虛擬環境
conda create -n Pytorch python=3.6
# 環境名稱:Pytorch python版本:3.6
2. 激活Pytorch環境
conda activate Pytorch
3. 在Pytorch環境下安裝torch
通過自己設備性能一一選擇,最後等到相應的代碼。將command複製在conda prompt 中運行。
4. 本地安裝
由於安裝過程中網絡的原因,在下載pytorch-0.4.1時,時間會很長,有時能會出現連接錯誤。所以將torch.whl文件下載到本地,分享給大家。
傳送門:【百度網盤】鏈接:https://pan.baidu.com/s/1pSofoGl8TN5iWM-VcZFWcw
提取碼:n3z7
【tips】:本地安裝whl文件,首先進入文件所在的當前目錄下通過運行
pip install torch-0.4.1-cp36-cp36m-win_amd64.whl
即可完成安裝。
5. 安裝torchvision
pip install torchvision
6. 測試
import torch # 導入torch包 如果沒有報錯則表示安裝成功
torch.cuda.is_available() # 返回True則表示GPU可用, 反之則不可用
記於2018.12.01