快速查看torch版本是否支持cuda:
python -c “import torch; print(torch.cuda.is_available()); print(torch.version)”
python -c "import torch; print(torch.cuda.is_available());
快速查看torch版本是否支持cuda:
python -c “import torch; print(torch.cuda.is_available()); print(torch.version)”
python -c "import torch; print(torch.cuda.is_available());