解决pytorch安装以后还是false

问题记录

使用pytorch官网的命令安装(直接看问题解决)
这里记录问题

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

import torch
print(torch.cuda.is_available()) # false
print(torch.__version__) # 2.0.0
print(torch.cuda.device_count()) # 0

解决办法

下载离线文件

镜像网站

cp 38 代表python版本38，linux 代表linux平台

linux服务器可以复制链接以后使用wget命令下载
注意需要下载torch torchvision torchaudio 对应的python版本

wget https://mirror.sjtu.edu.cn/pytorch-wheels/cu121/torch-2.1.0+cu121-cp39-cp39-linux_x86_64.whl

离线安装

cond的离线安装目录需要复制到本地仓库的pkgs文件夹下面，推荐直接使用pip

pip install --no-index --find-links=/home/backup/pytrochwheel torchaudio-2.1.0+cu121-cp39-cp39-linux_x86_64.whl

需要替换–find-links= 后面的目录到自己实际下载的目录
安装 torch torchvision torchaudio

结果

import torch
print(torch.cuda.is_available()) # true
print(torch.__version__) # 2.1.0+121
print(torch.cuda.device_count()) # 4