1.? ? 可以在cmd窗口輸入nvidia-smi命令來確認自己的顯卡是否有CUDA。

? ? ? ?沒有的話，需要安裝。（搜CUDA官網按照步驟來）

?2.? ??然后安裝pythorch。

? ? ? ? 必須安裝的有torch和torchvision兩個包，

? ? ? ? ！注意不要直接在Pycharm中自動下載安裝，

? ? ? ? 否則安裝的是CPU版本，無法進行GPU加速。

? ? ? ? 上官網：Start Locally | PyTorch? ?

? ? ? ?選擇適合自己電腦的版本。注意是conda安裝，不是pip。

? ? ? ?然后打開Pycharm或任意編譯器，執行以下代碼：

? ? ? ? ?import torch
? ? ? ? ?print(torch.cuda.is_available())
? ? ? ? ?print(torch.__version__)
? ? ??

? ? ?

? ? ? ? ?輸出 true 即可。（證明可以使用gpu進行加速）

? ? ?

? （詳情參考：Pytorch使用GPU加速的步驟_pycharm使用gpu加速_微笑小星的博客-CSDN博客）

?

總結

以上是生活随笔為你收集整理的Pytorch使用GPU加速的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。