import torch print(torch.__version__) print('gpu',torch.cuda.is_available())
如果输出false
解决办法安装cuda和cudnn
https://developer.nvidia.com/cuda-toolkit-archive
AssertionError: CUDA unavailable, invalid device 0 requested
最新推荐文章于 2024-05-12 18:01:27 发布
import torch print(torch.__version__) print('gpu',torch.cuda.is_available())
如果输出false
解决办法安装cuda和cudnn
https://developer.nvidia.com/cuda-toolkit-archive