Issue: torch.cuda.is_available() returns False

Cuda Info:
Run nvidia-smi:

Problem: It turned out that using conda will install a CPU-only version by running the conda command provided by  https://pytorch.org/get-started/locally/
Solution: Use the pip command to install the torch. In my case it should be
pip3 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
After the installation, check whether the gpu-based torch has been in

Issue: torch.cuda.is_available() returns False最先出现在Python成神之路

版权声明:
作者:zhangchen
链接:https://www.techfm.club/p/8022.html
来源:TechFM
文章版权归作者所有,未经允许请勿转载。

THE END
分享
二维码
< <上一篇
下一篇>>