pytorch——gpu版本安装

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

最后编辑于 ：
©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容