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里运行程序,可以了

 

posted @   _海阔天空  阅读(1884)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!
点击右上角即可分享
微信分享提示