美文网首页
Windows 安装 pytorch

Windows 安装 pytorch

作者: katsura_911 | 来源:发表于2018-01-25 01:09 被阅读0次

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

    相关文章

      网友评论

          本文标题:Windows 安装 pytorch

          本文链接:https://www.haomeiwen.com/subject/hkapaxtx.html