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.验证

猜你喜欢

转载自blog.csdn.net/yinmengtingjoy/article/details/85000045
今日推荐