pytorch环境搭建

使用Anaconda新建环境

更新pip版本

python -m pip install --upgrade pip

安装pytorch

1. 获取GPU的CUDA版本。

打开终端，输入nvidia-smi。

2. 进入官网获取pytorch安装指令。

引用站外地址

Pytorch最新版

下载链接

引用站外地址

Pytorch历史版

下载链接

3. 安装指定Cuda版本的Pytorch

在Ubuntu中安装Cuda-11.8版本的Pytorch-2.1.1。

从Pytorch历史版中找到对应的指令。

pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 --index-url https://download.pytorch.org/whl/cu118

尽量使用pip进行Pytorch安装，使用conda安装可能出现卡住的情况