先贴几个参考的帖子:
Ubuntu20.04下深度学习环境配置(持续维护)_ubuntu深度学习-CSDN博客
前置内容如换源、GPU驱动、gcc、anaconda等都已配置好,只写怎么在已有环境下搭建新的云·深度学习代码运行环境
1、创建新环境:conda create -n litimono python=3.8
【ps:conda activate litimono,激活环境,若失败,则使用source activate litimono激活环境】
2、安装pytorch:conda install pytorch==1.11.0 torchvision torchaudio cudatoolkit=10.2 -c pytorch
【此处根据自己的需要做更改】
验证:
$python \\进入python执行界面
$import torch \\若无任何提示,安装成功
$print(torch.__version__) \\查看pytorch版本,出现1.11.0则表示安装成功
$exit() \\退出python执行界面
3、安装cuda,若之前已经配置过环境,cuda应该是安装好了的,在环境下使用nvcc --version检查,若出现下图则表明cuda安装成功,否则参考上述链接安装cuda
4、安装cudnn同上,若之前已经配置过环境,cudnn应该是安装好了的,在进行完 “2、pytorch安装” 后用torch.cuda.is_available()进行验证的,若出现Ture则表示安装成功,否则参考上述链接安装cudnn。
5.litimono复现过程
litemono代码_linear_warmup_cosine_annealing_warm_restarts_weigh-CSDN博客
在上面的步骤完成后,进行以上文章的过程5--在终端使用
python -c "import site; print(site.getsitepackages())
来确定文件位置,之后将下载解压的文件夹按文章说明放到该位置即可。