筆者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