基于Anaconda搭建Pytorch环境

时间:2024-04-27 12:15:36

准备虚拟环境

创建一个虚拟创建:

conda create --name nlp python=3.11.7

激活虚拟环境:

conda activate nlp

安装pytorh

首先,可以通过任务管理器查看你的电脑是否支持GPU:
在这里插入图片描述

如果支持,到网址:https://pytorch.org/get-started/locally/#windows-pip 选择CUDA:

在这里插入图片描述

得到安装命令如下:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

如果不支持,则选择CPU:

在这里插入图片描述

得到的安装命令如下:

pip3 install torch torchvision torchaudio

根据自己的网络情况,可能需要下载比较久的时间:
在这里插入图片描述