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里运行程序,可以了
分类:
pytorch
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 全程不用写代码,我用AI程序员写了一个飞机大战
· DeepSeek 开源周回顾「GitHub 热点速览」
· 记一次.NET内存居高不下排查解决与启示
· 物流快递公司核心技术能力-地址解析分单基础技术分享
· .NET 10首个预览版发布:重大改进与新特性概览!