【Pytorch-gpu版安装教程，针对cuda11.1】

iiidep

已于 2022-11-29 12:31:04 修改

阅读量3k

点赞数

CC 4.0 BY-SA版权

文章标签： pytorch python 深度学习

于 2022-11-29 10:28:05 首次发布

本文链接：https://blog.youkuaiyun.com/iiidep/article/details/128092746

本文详细介绍了在CUDA 11.1环境下，如何离线安装PyTorch-GPU版本。通过检查NVIDIA显卡版本，从官方下载页选择匹配的whl文件，并在Anaconda Prompt中进行安装。过程中遇到了conda解决环境失败、选择正确操作系统版本等问题，最终成功验证安装并确保GPU支持。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

针对cuda11.1版本安装pytorch-gpu版本@TOC

配了3天的环境终于true了，简单列举一下这三天遇到的问题和解决方法。
首先介绍一下成功的路径。
由于校园网不好，我使用了离线下载。
在下载torch之前，最好先打开命令行（输入cmd）,在命令行中输入nvidia-smi查看显卡对应的版本：
在这里插入图片描述
然后打开这个网址https://download.pytorch.org/whl/torch_stable.html，寻找版本对应的torch和torchvision进行下载。
此处介绍gpu版本，因为有英伟达显卡，所以选择下载gpu版本，此处我下载的是这两个
cu111/torch-1.9.0%2Bcu111-cp38-cp38-win_amd64.whl
cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-win_amd64.whl
（cu111是对应cuda11.1），选择windows版本而不是linux，因为这里我有遇到过，先下载了linux版本然后报错，因为文件里只能识别win_amd64类似的名字。所以要选对版本。
我是将它们下载到了D盘的安装包里，然后在anaconda prompt中首先要进入存放这两个包的地址中，然后新建虚拟地址
在这里插入图片描述
最后输入python,import torch若未报错，则安装成功，输入torc