win11下利用conda进行pytorch安装-cuda11.6-泛用安装思路

第一剑柄

已于 2023-02-21 11:26:04 修改

阅读量1.3k

点赞数 1

文章标签： pytorch conda 深度学习

于 2023-02-21 11:13:57 首次发布

本文链接：https://blog.csdn.net/qq_34740266/article/details/129137794

版权

安装pytorch！

自学习深度学习以来，安装过几次pytorch，但是每次安装的过程都不太相同。安装教程有一定的时效性，本文章会给出11.6cuda的torch安装，同时会给出一个torch的安装思路（即便在你看到文章的时候可能torch官网已经更新了）

首先去torch的官网查看现在是用的什么什么版本的cuda，https://pytorch.org/
看看compute platform
然后根据这个去下载cuda。
https://developer.nvidia.com/cuda-11-6-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=11

再下载一个对应版本的cudnn，https://developer.nvidia.com/rdp/cudnn-archive
cudnn解压到cuda目录里，比如下图（具体要看你安装哪里了）：
cudnn解压位置
我建议你先访问官网给的 --extra-index-url 网址，查看一下里面的包，比如下图：
在这里插入图片描述
7
这个时候你可以开始conda -n newenv python=3.7； conda一个环境出来。

最后按照torch命令安装，比如：
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

这个过程你会发出现下载缓慢导致链接断开等情况。这里可以参考 https://blog.csdn.net/fns212/article/details/128459862
这一篇博客。
具体操作为访问官网给的 --extra-index-url 网址，自己下载对应的包。比如我需要的是py3.7、cuda116的包。你在输入上述命令时会发现你需要的包，自己去下载后，放入conda的的pkgs包里面，重新执行命令。最后可以安装成功。

事实上，安装后，我还遇到了其他问题，如import torch出现numpy版本问题相关报错，采用卸载numpy重安装来解决。这些可以查找其他相关教程。