pytorch环境搭建

使用Anaconda新建环境

更新pip版本

python -m pip install --upgrade pip

安装pytorch

1. 获取GPU的CUDA版本。

打开cmd终端，输入nvidia-smi。

2. 进入官网获取conda指令。

引用站外地址

pytorch

下载链接

使用conda比pip安装较快。

3. 例如Windows下CUDA12.1版本，执行如下指令：

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

测试

import torch

x = torch.rand(5, 3)
print(x)