admin 管理员组

文章数量: 1184232

在此之前,需做好PyCharm、Python、CUDA、cuDNN安装

PyCharm、Python、CUDA、cuDNN安装教程【汇总合集】-CSDN博客

第一步:下载 torch 和 torchvision 包

地址:https://download.pytorch/whl/torch_stable.html

Pytorch 和 CUDA 和 Python 版本要对应

torchvision 和 CUDA 和 Python 版本要对应

按Ctrl+F搜索

例如:cu118-cp39

含义介绍:torch-2.0.1表示torch版本2.0.1,

                  cu118表示CUDA版本11.8(必须小于等于电脑CUDA版本),

                &n

本文标签: 方法 GPU pycharm TorchVision pytorch