windows/ubuntu+anaconda安装PyTorch

版本:

window10 64位

Anaconda 4.5.11

  • 在官网https://pytorch.org/选择要安装的版本,获取安装命令。使用conda命令下载非常慢,所以用了pip命令。安装的pytorch版本为1.0.0

windows/ubuntu+anaconda安装PyTorch_第1张图片

验证:

windows/ubuntu+anaconda安装PyTorch_第2张图片

.成功!


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

windows/ubuntu+anaconda安装PyTorch_第3张图片
1.新建环境mytorch

conda create -n mytorch python=3.6
source activate mytorch

进入环境目录

2.找到相应版本的安装命令

windows/ubuntu+anaconda安装PyTorch_第4张图片

 conda install pytorch torchvision -c pytorch

windows/ubuntu+anaconda安装PyTorch_第5张图片

3.验证

windows/ubuntu+anaconda安装PyTorch_第6张图片

你可能感兴趣的:(pytorch)