pytorch安装问题解决

背景描述：

直接从官网用pip或者conda命令下载gpu版本的pytorch经常会遇到安装缓慢，但是换源后安装成功的pytorch是cpu版本的，查看指令：

import torch
print(torch.__version__)
print(torch.cuda.is_available())

如果是gpu版本torch.__version__的输出内容应该带有cuxxx，xxx为你的cuda版本，但是有时候会不显示，这时候可以用最底下那行命令查看，如果输出是true则代表安装成功

解决方案：

下载.whl文件到本地，然后从本地进行安装

网址：download.pytorch.org/whl/torch_stable.html

具体流程：

1. 确定你需要下载的版本，然后control f进行查找，如我需要在linux操作系统中下载cuda版本为12.1的pytorch:

cp是你对应的python环境版本，如3.11就是cp311

2.下载

windows或者有图形化界面的linux中直接点击对应的连接即可

服务器中使用wget命令，具体为：

wget https://download.pytorch.org/whl/

在这之后加上你要下载的名字，完整的示例为：

wget https://download.pytorch.org/whl/cu121/torch-2.3.1%2Bcu121-cp311-cp311-linux_x86_64.whl

3.安装

找到你的文件下载地址，然后在命令提示行，进入虚拟环境中进行安装，如：

conda activate DDPO
pip install torch-2.3.1+cu121-cp311-cp311-linux_x86_64.whl

注意：我现在是在.whl文件的目录下直接进行安装的，如果不是在文件所在目录中则需要在pip install后面加上文件的完整路径

结果展示：