Torchvision Cuda Problem.md

记录一下 torchvision 遇到的问题

1
torch.cuda.DeferredCudaCallError: CUDA call failed lazily at initialization with error: device >= 0 && device < num_gpus INTERNAL ASSERT FAILED at "/opt/conda/conda-bld/pytorch_1712608839953/work/aten/src/ATen/cuda/CUDAContext.cpp":50, please report a bug to PyTorch. device=, num_gpus=

我调整了代码，遇到了这个问题，结果是因为引入了 from torchvision import transforms。

这个问题可能是 nvidia 驱动问题导致的。解决这个问题，最好的方法是改一下 torch 的版本，找到一个和驱动接近的。

重新安装 nvidia 驱动付出的代价比较大。