(仅个人学习记录使用)
1.下载anaconda3和安装
保存在 /project/username/ 文件下
cd /project/username/
sh Anaconda3-2022.05-Linux-x86_64.sh
中间更换路径 ,不能安装在home里,内存不够
/project/username/anaconda3
2.创建pytorch虚拟环境
在指定目录下创建:
conda create --prefix=/project/username/anaconda3/envs/pytorch python=3.8
否则可能安装在home文件里,会报内存错误
3.安装torch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch4
4.修改预训练模型保存位置
找到hub.py 文件所在位置
/project/username/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch
打开hub.py文件
定位值hub_dir
模型的保存位置:
hub_dir=‘/project/username/anaconda3/envs/pytorch/lib/python3.8/site-packages/torch’
总之,不能安装到home下,不然会报各种容量的错误