美文网首页
安装PyTorch

安装PyTorch

作者: ClarenceHoo | 来源:发表于2019-03-28 13:22 被阅读0次

    参考资料:
    pytorch中文手册:https://github.com/zergtant/pytorch-handbook
    因为本机已经安装过tensorflow了,故而在虚拟环境中安装,已经安装好anaconda3

    #pytorch为环境名,这里创建python3.6版。
    conda create -n pytorch python=3.6
    # 切换到pytorch环境
    activate pytorch
    # ***以下为1.0版本安装***
    #安装GPU版本,根据cuda版本选择cuda80,cuda92,如果cuda是9.0版,则不需要
    #直接conda install pytorch -c pytorch 即可
    # win下查看cuda版本命令nvcc -V
    conda install pytorch cuda92 -c pytorch
    # cpu版本使用
    # conda install pytorch-cpu -c pytorch
    
    
    # torchversion提供了conda的安装包,可以用conda直接安装了
    # 安装方式如下:
    # cpu版本
    conda install pytorch-cpu torchvision-cpu -c pytorch
    # GPU版 
    conda install pytorch torchvision cudatoolkit=9.2 -c pytorch
    # cudatoolkit后跟着相应的cuda版本
    

    相关文章

      网友评论

          本文标题:安装PyTorch

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