torch.cuda.available()
对于torch.cuda.available()出现False的解决方法：检查版本是否正确我的cuda是11.4pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html检查代码是否正确device =
·
对于torch.cuda.available()出现False的解决方法：
- 检查版本是否正确
我的cuda是11.4
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
- 检查代码是否正确
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") # 在cpu还是gpu上训练
- 以上都正确，将conda安装改为pip安装
为开发者提供学习成长、分享交流、生态实践、资源工具等服务，帮助开发者快速成长。
更多推荐
0
0- 0
已为社区贡献1条内容
所有评论(0)