首页 > 技术文章 > pytorch 无法调用cuda

zkwarrior 2022-02-07 21:58 原文

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

 

推荐阅读