🚀本文選自專欄：《小白入門深度學習》
🚀推薦精選專欄：《深度學習100例》
文章目錄
- 簡單的方法：
- 快捷的方法
- 檢查是否安裝成功
簡單的方法：
進入 【官網】，通過官方提供的方法進行安裝，這個方法的好處是簡單
快捷的方法
上面的方法屬于pip 在線安裝，比較方便，但是安裝速度會慢很多，如果希望更快，建議先將安裝包檔案下載到本地，然后在本地進行安裝， 具體方法如下：
第一步：下載安裝包檔案（.whl檔案）
- 按照上面的方法獲取命令A
pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html， WIN鍵+R輸入cmd打開命令列，并運行命令A
- 在https://download.pytorch.org/whl/torch_stable.html頁面中找到上面下載的檔案（即安裝包檔案）
cu111/torch-1.9.0%2Bcu111-cp36-cp36m-win_amd64.whl
第二步：直接下載.whl檔案
第三步：使用pip進行本地安裝
檢查是否安裝成功
運行下面兩份代碼，第一份代碼運行成功說明你的PyTorch安裝成功，第二份代碼運行成功并回傳True說明你GPU版本的PyTorch安裝成功，
import torch
x = torch.rand(5, 3)
print(x)
import torch
torch.cuda.is_available()
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/296480.html
標籤：AI