Pytorch开发环境搭建（windows、conda）

环境说明

系统：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())