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