windows下pytorch安裝

前提安裝好CUDA和cuDNN 如果沒有安裝好CUDA和cuDNN請轉移至 https://blog.csdn.net/suyongcai1234/article/details/90750940

pytorch安裝

進入pytorch官網：https://pytorch.org/get-started/locally/

選擇對應的CUDA版本

按照提示的pip指令在windows終端中運作即可

測試

在python代碼中

import torch

import torchvision

沒有報錯即為測試成功