PyTorch教程-0:PyTorch的安裝

筆者PyTorch的全部簡單教程請訪問:http://www.lxweimin.com/nb/48831659

PyTorch教程-0:PyTorch的安裝

安裝PyTorch

安裝PyTorch參考如下的頁面:https://pytorch.org/get-started/locally/

根據自己電腦的系統、安裝方式、CUDA版本、PyTorch版本等進行配置選擇,然后復制下邊的命令到命令行工具執行即可

1.png

對于CUDA版本,這里介紹Windows系統的查看方式:

  • 在桌面右鍵選擇 NVIDIA 控制面板,或者直接搜索“NVIDIA control panel”,再或者在右下角收起的程序中尋找。打開控制面板后選擇上方選項中的最后一個“幫助”下的系統信息。
4.png
  • 在系統信息中,切換到“組件”選項卡,找到“NVCUDA64.DLL”,后邊的“產品名稱”中可以看到其版本
2.png

將獲取到的命令復制到命令行中執行,等待一段時間后即可安裝成功。

3.png

驗證是否安裝成功

在一個Python解釋器中通過如下的代碼驗證是否安裝成功(以及CUDA是否可用):

import torch

torch.rand(4,3)

torch.cuda.is_avaliable()
5.png
最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容