1、创建虚拟环境
conda create --name '名字' python='python版本号'
2、进入虚拟环境
conda activate '名字'
3、确认自己要安装的torch版本和torchvision版本,进入https://pytorch.org/get-started/previous-versions/ 复制相应的命令,运行即可
注:建议使用pip命令,我使用conda命令安装失败
例:pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
4、运行完毕应该安装成功,可以检查一下,检查方式是进入虚拟环境后输入python命令,接着输入import torch命令后,输入torch.cuda.is_available(),如果结果是true,则安装成功。
5、安装剩余的包可以通过直接运行代码,看代码报错提示缺少什么库,再进行对应的安装,可以使用如下命令(带清华镜像源)
pip install 库名 -i https://pypi.tuna.tsinghua.edu.cn/simple
注:有一些库提示没有,比如yaml库,下载的时候是pip install pyyaml ,类似的如果直接安装报错,那么进行百度即可