1. 程式人生 > >windows/ubuntu+anaconda安裝PyTorch

windows/ubuntu+anaconda安裝PyTorch

版本:

window10 64位

Anaconda 4.5.11

  • 在官網https://pytorch.org/選擇要安裝的版本,獲取安裝命令。使用conda命令下載非常慢,所以用了pip命令。安裝的pytorch版本為1.0.0

驗證:

.成功!


ubuntu+anaconda安裝PyTorch-GPU

版本:

Anaconda 4.5.4

nvidia-smi檢視顯示卡資訊

nvcc -V檢視cuda資訊:Cuda compilation tools, release 9.0, V9.0.176
另:

cuda 版本 
cat /usr/local/cuda/version.txt

cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2


1.新建環境mytorch

conda create -n mytorch python=3.6
source activate mytorch

進入環境目錄

2.找到相應版本的安裝命令

 conda install pytorch torchvision -c pytorch

3.驗證