Anaconda下安裝Pytorch-GPU版

設備環境:

  1. windows10  
  2. Anaconda5.2
  3. 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


Pytorch開始頁

 通過自己設備性能一一選擇,最後等到相應的代碼。將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

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