linux-Anaconda+conda创建python-pyt
2018-05-03 本文已影响0人
水球喵
创建Anaconda到python的步骤从https://www.jianshu.com/p/a37538f46551这里查看。
cat /usr/local/cuda/version.txt查看cuda版本
https://pytorch.org/从这里下载对应版本的pytorch
image.png
好了,发现它将tensorflow版本更新到了1.7
然后import torch
不报错就成功了
如果想要安装老版本的pytorch(http://pytorch.apachecn.org/previous-versions/)
例如:我想安装cuda9的pytorch=0.2.0版本
conda install pytorch=0.2.0 torchvision -c pytorch