1 问题
电脑拥有独立显卡,pycharm中torch.cuda.is_available()总是返回false。
2 方法
首先在cmd当中输入NVIDIA-smi查看当前CUDA的版本,再到torch官网下载对用的torch版本。
3 结语
针对CUDA版本低于11.3,没有相应的下载指令,下载其他版本的Pytorch依旧返回False,则应当对GPU的CUDA的版本进行升级。到NVIDIA GeForce Experience对当前GPU进行升级之后,再到官网下载对应的torch。
本文分享自 算法与编程之美 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!