使用GPU计算python程序，如何正确安装cuda版本的torch模块，详细教程

1. 查看电脑可以安装的cuda版本

方法一：右击打开NVIDIA控制面板，点击：帮助——系统信息——组件

此电脑可安装的cuda版本为11.4.94

方法二：也可在cmd中输入nvidia-smi，回车

需要关注两个地方：

1）显卡驱动版本：471.41

2）显卡支持最高的CUDA版本：11.4

2. 选择cuda版本的pytorch

进入pytorch官网：https://pytorch.org/get-started/locally/

选择目前可安装的pytorch的稳定版本，但对于作者来说，目前最新版本是不能安装的，大家可以下载自己需要的版本。

pytorch也提供了所有的版本，可自行下载：https://pytorch.org/get-started/previous-versions/

下面以官网推荐的稳定版本介绍安装过程，其他版本同理：

官网会提供pip安装命令：pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117，也可以选择其他安装方式。

3. 进入cmd安装pytorch

在python的Scripts文件夹下运行cmd，输入：pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117，利用pip模块下载相应pytorch版本和进行安装。

4. 如果下载过慢，或者安装不成功，也可根据下载路径自行下载，然后用pip进行安装

根据下载的---.whl文件，利用pip进行模块安装，可参考博客：https://blog.csdn.net/weixin_44762762/article/details/129561289?spm=1001.2014.3001.5501中的第二种安装方式

下面截图为https://blog.csdn.net/weixin_44762762/article/details/129561289?spm=1001.2014.3001.5501博客中的安装方式说明

5. 在pycharm中查看安装的torch版本

或者通过python的IDLE shell查看

——————————————————————

希望此博客能对您有所帮助，如果仍有问题，欢迎留言讨论！！