win10+pytorch+cuda安裝

1、查看cuda版本 打開nvidia控制面板,點擊系統信息-->組件-->查看NVCUDA.DLL那一項。可以看到我的cuda是10.1版本的。如果沒有的話,可以先去安裝cuda,這裏不再介紹cuda的安裝方法。 2、下載pytorch。官網鏈接https://pytorch.org/。根據自己的具體情況進行選擇,會生成一個相應的安裝命令 3、用上一步自己選擇後生成的命令進行安裝 4、判斷是否安
相關文章
相關標籤/搜索