【Python学习】pytorch GPU版本安装

CUDA 安装和cudnn安装看网上的帖子就可以

我的版本:
image

链接:https://pan.xunlei.com/s/VN8JDNhpnnKqPGXTtFK8guOeA1?pwd=2bzt#
提取码:2bzt
没什么难度

pytorch安装

就这块我弄了快一天了,整了好久才弄好

简单记录一下,方便以后再次安装

版本链接,点进去这个界面
image
选择他给的命令就可以

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

打开Anaconda Powershell Prompt

创建一个虚拟环境

conda create -n pytorch python=x.x
conda create -n your_env_name python=x.x

进入到虚拟环境中

conda activate pytorch

在虚拟环境中输入上面的指令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

-c 后面是下载pytorch torchvision torchaudio cudatoolkit=11.3这几个东西的路径,服务器在国外,墙内下的失败概率大,所以可以改一下路径,我用的南京大学的镜像网站。
南京大学镜像网站
链接:
https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/

更改后的指令
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c https://mirror.nju.edu.cn/pub/anaconda/cloud/pytorch/win-64/

验证是否安装成功

先进入之前创建的虚拟环境
conda activate + 虚拟环境的名称
例子
conda activate pytorch
然后输入下面内容
image

posted @ 2022-07-31 20:04  寥若辰星  阅读(222)  评论(0编辑  收藏  举报