Windows+pytorch,百度了半天都没有安装好,然后google了五分钟安装好了,秀的我头晕,还是自己的脑子不太聪明啊。
先上安装方法:
环境:Anaconda 5.0.1 ,python 3.6
安装步骤:1.在Anaconda下新建一个环境,在root下输入命令
conda create -n pytorchenv //创建一个新的环境pytorchenv
activate pytorchenv //激活pytorchenv'
2.安装pytorch
conda install -c peterjc123 pytorch
安装完成后测试,import torch成功。
国内帖子的安装方法是你自己去下一个tar.gz的安装包,然后安装,虽然可以安装成功,但是在导入torch包会报错:
from torch._C import *
ImportError: No module named _C
据说解决方案是不要从torch根目录运行此命令,但是我是使用pip命令安装的包,用python命令来import,这个import命令还有运行目录的说法吗?我使用jupyter 来import同样会报该错误,ipynb文件的目录总不能是torch的根目录吧?这个问题希望能有大佬指点。
现在好像官方已经发布了安装包,所以直接通过命令安装就可以了、
ps:因为我的电脑是A卡,安装好pytorch之后是不能使用gpu加速的,考虑换个显卡?
测试命令:
import torch
print(torch.cuda.is_available())