1.下載最新版本的pytorch

https://pytorch.org/官網(wǎng)可以根據(jù)操作系統(tǒng)，選擇cuda版本， 和選擇conda命令還是pip命令:

例如下載最新版本的pytorch命令為:

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

2.pytorch歷史版本下載

網(wǎng)址:https://pytorch.org/get-started/previous-versions/

3.下載完成后檢測(cè)gpu是否可用

代碼:

import torch torch.cuda.is_available()

輸出為ture表示gpu可用
只是檢測(cè)CUDA是否安裝正確并能被Pytorch檢測(cè)到，并沒(méi)有說(shuō)明是否能正常使用，要想看Pytorch能不能調(diào)用cuda加速，還需要簡(jiǎn)單的測(cè)試一下：

a=torch.Tensor([1,2]) a=a.cuda() a

總結(jié)

以上是生活随笔為你收集整理的pytorch历史版本下载的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。