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对应版本

上一篇下一篇

猜你喜欢

热点阅读