Pytorch GPU 训练环境搭建_torch cu101

作者：Gausst松鼠会 | 2024-06-03 12:59:17

踩

torch cu101

之前⼀直使⽤ Tensorflow 训练模型，第⼀次训练Pytorch模型的时候，发现速度很慢，仔细观察，发现GPU 内存占⽤为0，基本没有使⽤GPU。


AssertionError: CUDA unavailable, invalid device 0 requested
cuda不可⽤报错，现实没有有效的驱动可使⽤

测试cuda是否配置正确


import torch
print(torch.cuda.is_available())

重新安装cuda

检测本地GPU CUDA版本 nvidia-smi


pip3 install torch1.9.0+cu101 torchvision0.10.0+cu101 torchaudio=0.9.0 -f
https://download.pytorch.org/whl/torch_stable.html

当前 torchvision0.10.0+cu101 版本必须是匹配的。如果版本不匹配，如上⾯的命令，则会出现错误

我们打开网站

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/667345