问题:
如果和下图一样,print(torch.cuda.is_available()),结果是false:
说明:所使用的环境是anaconda pytorch。
你打算使用GPU来加速某个项目,但是GPU不能用,原因是没有安装GPU版本的torch和torchvision。
使用pip安装GPU版本的支持CUDA 11.7的torch和torchvision,并且版本分别为torch==2.0.1和torchvision==0.15.2。具体可以根据需要的版本进行修改。
pip install torch==2.0.1+cu117 torchvision==0.15.2+cu117 -f https://download.pytorch.org/whl/torch_stable.html
安装完成之后,检查是否可以正常使用。
输入python,回车,然后输入:
import torch
print(torch.cuda.is_available())
结果:
这样就完成了,可以使用GPU来加速项目。