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