pytorch 无法调用cuda


切换到pytorch环境下,调用cuda都是false,无论是命令行下还是pycharm下,如下图

第一反应是cuda版本太高了??搜了一下,发现可以支持啊

 所以应该不是cuda版本的事,就在base环境下试了一下,发现在base环境下显示有GPU,说明cuda没问题,那就是pytorch环境下的安装问题,可能没注意pytorch安装成cpu版本的了。

 重新在pytorch环境下运行命令:conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

 

-c 即 -channel
频道是Navigator和conda查找包的位置,(source)具有相同名称的包可能存在于多个通道上,如果希望从默认通道以外的其他通道安装,则指定要使用哪个通道的一种方法是使用 conda install -c channel_name package_name语法。

安装完成再在pycharm里运行程序,可以了

 


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM