Pytorch-1.0.1 深度学习环境踩坑备忘录

最新推荐文章于 2024-03-26 12:49:31 发布

原创

最新推荐文章于 2024-03-26 12:49:31 发布 · 764 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch1.0.1 GPU安装

这篇博客记录了在Windows 10上安装Pytorch 1.0.1遇到的问题及解决方案，包括更换国内源以解决下载速度慢、下载失败后的离线安装方法、GPU测试失败的显卡驱动更新，以及cuDNN文件的正确放置。

Pytorch-1.0.1 深度学习环境踩坑备忘录

安装环境：windows 10 x64

显卡：nvidia 940mx

pytorch参考文章：https://zhuanlan.zhihu.com/p/53652964

Question 1 国外源太慢，切换为国内源

两种办法，解决国外源问题：命令行或修改.condarc文件

cmd 输入：

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

修改文件

channels:
  - https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
show_channel_urls: true
ssl_verify: true

实际不用copy - https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/</