情景是:明明pip install torch>=1.10 安装成功了,但conda环境下运行
print(torch.cuda.is_available())
却得到
False
后来通过
print(torch.__version__)
确实是安装提CPU版本而非GPU版本
找到以前另一个安装笔记使用以下安装解决
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
但正确的处理方法应该是Pip手工安装 ,指定清楚,比如以下demo
pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio===0.7.2
速度慢还可以用迅雷下载实际文件后 直接安装。
参考自:https://blog.csdn.net/qq_46126258/article/details/112708781
本文《conda环境下print(torch.cuda.is_available()) False》由爱思考吧 isres.com 分享,转载请注明出处。本文网址:https://www.isres.com/jingyan2/57.html
本站主要收集测评能够节省时间和提升效率的软件工具并分享使用与学习的过程和感受,任何情况下用户都需遵守所使用软件资源的相关协议。与《conda环境下print(torch.cuda.is_available()) False》有关的本站文章及资源仅供个人学习交流使用,相关资源请下载后24小时内删除,请勿用于其它用途,因此产生的任何问题由您自行承担。