美文网首页
pytorch——gpu版本安装

pytorch——gpu版本安装

作者: 龙鹰图腾223 | 来源:发表于2021-02-05 16:54 被阅读0次

    1、检查显卡驱动(控制面板-小图标-英伟达,帮助-信息),安装和显卡对应的cuda版本,添加环境变量

    用cudann里面的文件替换,参考地址:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

    (可以在没有conda环境下直接使用)

    建新的anaconda环境

    (检查显卡驱动是否满足cuda版本,torch版本 和 cuda版本需要对应)

    (可以在清华源下载离线安装包,https://mirrors.tuna.tsinghua.edu.cn/help/pypi/)

    分步安装,注意区分linux 和 win版本

    【1】https://www.cnblogs.com/lyl0618/p/13323010.html    torch& torchvision对应版本

    相关文章

      网友评论

          本文标题:pytorch——gpu版本安装

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