摘要: 检查CUDA相关配置 检查GPU驱动 nvidia-smi 检查CUDA版本 nvcc -V 若未安装CUDA会报错,若已安装会显示CUDA版本。 检查Pytorch是否可用 import torch torch.cuda.is_available() 若运行失败则torch未安装,若显示True则 阅读全文
posted @ 2024-11-29 17:54 半塘加冰 阅读(32) 评论(0) 推荐(0) 编辑
点击右上角即可分享
微信分享提示