Windows+pycharm安裝pytorch

最近要參加一個比賽,需要用到深度學習的框架,老師建議我們用pytorch,但是安裝的時候總是發現各種錯誤。 解決方案是:去pytorch官網生成下載的命令,然後再cmd上執行 這裏我們發現有CUDA的選項,這就需要我們安裝cuda 怎麼安裝cuda? 首先我們要知道我們的獨立顯卡的型號: 然後我們要知道我們的顯卡是否支持cuda:點擊這裏    發現是支持的,算力爲6.1: 接下來就要下載cuda
相關文章
相關標籤/搜索