pytorch 的安裝

安裝步驟 1.檢查是否有合適的GPU, 若有安裝Cuda與CuDNN 2.下載pytorch安裝文件,進行pytorch的安裝 檢查電腦是否有合適的GPU 在桌面上右擊如果能找到NVIDA控制面板,則說明該電腦有GPU。控制面板如下,並通過查看系統信息獲取支持的Cuda版本 點擊 幫助->點擊 系統信息 彈出下面的對話框,在驅動程序版本那一欄就能看到該計算機使用的驅動版本。   下載Cuda 官網
相關文章
相關標籤/搜索