下载GPU版本Pytorch

参考小土堆视频下载

27. GPU版本-安装 PyTorch（方法2）_哔哩哔哩_bilibili

命令行查看本设备的cudaDriver版本

nvidia-smi

查看本驱动能够支持最大版本cuda:

在pytorch官网下载对应版本

打开cmd

conda activate pytorchName

方法一：直接安装

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

方法二：本地安装

复制下载地址，通过迅雷快速下载到本地

cmd下载

(pytorch) C:\Users\MyLENOVO>pip install address

下载结束后再次执行

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

下载剩余的包

下载成功即可完成

验证：

进入pytorch环境,

(pytorch) C:\Users\LENOVO>python
>>> import torch
>>> torch.cuda.is_available()
True

完成