Windows 安装 pytorch

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())

    原文作者:katsura_911
    原文地址: https://www.jianshu.com/p/63147012fa2c
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞