pytorch的cuda環境搭建(GPU版本安裝)

第一步:安裝cuda 安裝cuda,這裏還須要安裝對應版本的Visual Studio(具體參考本博主的博客 https://blog.csdn.net/qq_36653505/article/details/81368346 )web 第二步:安裝cudnn cudnn是pytorch搭建深度學習模型的依賴,沒有它,不能運行卷積等操做。 下載對應版本的cudnn連接:https://develo
相關文章
相關標籤/搜索