问题描述

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

使用了这个命令之后

import torch  
  
print(torch.cuda.is_available())#输出为false

输出仍然为false

解决办法

使用这两个命令将

pip uninstall torch torchvision torchaudio filelock typing-extensions sympy networkx jinja2 fsspec mkl numpy pillow

pip cache purge

下载相关的包彻底删除，第一个命令是移除包，第二个命令是清除缓存

重新使用，注意这里一定不能使用清华的源，不然仍然会是false

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

命令下载，如果第一个下载的包大小为2.4g就可以print(torch.cuda.is_available())输出就为true了
但是如果第一个下载的包大小为150mb左右，输出就仍然会是false，原因是使用了清华的源，会下载轻量版的torch
所以不要再后面加上清华源进行下载

错误命令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121 -i https://pypi.tuna.tsinghua.edu.cn/simple

正确命令

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

总结

根本原因是使用了清华源的轻量版torch导致的问题。