准备虚拟环境
创建一个虚拟创建:
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
根据自己的网络情况,可能需要下载比较久的时间: