pytorch——gpu版本安装
2021-02-05 本文已影响0人
龙鹰图腾223
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对应版本