环境说明
系统：windows 11
IDE： pycharm
工具：Anaconda
安装conda
到官网下装对应的版本安装即可
https://www.anaconda.com/download
安装后创建一个新的python环境
安装pytorch
按官网指引 https://pytorch.org/get-started/locally/
要先在cmd上检查一下自己的显卡CUDA版本，然后选上面对应的CUDA版本
启动上面创建的环境后，执行下面的pip3命令安装即可
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
测试效果
配置IDE
新建一个脚本文件，并执行相关的测试代码
代码如下
import torch
print(torch.__version__)
print(torch.cuda.is_available())