assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested' # check availability
AssertionError: CUDA unavailable, invalid device 0 requested
okey 此时此刻又到了经典cuda不可用环节,
建了一个py文件,测了一下
import torch
print(torch.__version__)
print(torch.cuda.is_available())
torch下载好了,但是cuda不可用,我感觉是因为这个版本2.2.2太高的原因。然后我根据
之前写的这个文章的备注部分直接改了,就是因为cuda对不上号的原因,我在终端输入这个代码等待安装就行了,如下图。
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
PS:这个不是直接复现的别人的论文吗,requestment里边先安装了一波,安装完之后出现的这个问题,直接找对应的安装命令行就行了,不用跟那个文章里一样下载那么多。
然后我再验证了一下 ,显示true就行了。