assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested' # check availability AssertionError: CUDA unavailable, invalid device 0 requested
搞了一下午,发现下载的是cpu版本
镜像下载自动就是cpu版本
assert torch.cuda.is_available(), f'CUDA unavailable, invalid device {device} requested' # check availability AssertionError: CUDA unavailable, invalid device 0 requested
搞了一下午,发现下载的是cpu版本
镜像下载自动就是cpu版本