一、Anaconda环境安装
- 先下载Anaconda Linux版,并将其重命名为anaconda2020.sh
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2020.07-Linux-x86_64.sh --header="User-Agent: Mozilla/5.0 (Windows NT 6.0) AppleWebKit/537.11 (KHTML, like Gecko) Chrome/23.0.1271.97 Safari/537.11" --header="Referer: https://codeclips.cc/" -O anaconda2020.sh
- 安装Anaconda一路点击yes和确认即可。
bash anaconda2020.sh
- 添加环境变量
a. vim ~/bashrc //打开环境变量配置文件
b. 在文件中,找个空行,填写以下语句,添加环境变量
export PATH=/home/你的用户名/anaconda3/bin:$PATH
c. 保存文件并退出。
d. source ~/bashrc //刷新
e. conda --version //查看版本
- 创建Python3.9虚拟环境,并进入。
a. conda create -n 3D-3.9 python==3.9 //-n指环境名称,3D-3.9为名称,可自定义。
b. y
c. conda activate 3D-3.9 //进入环境
此外
conda deactivate //退出环境
conda env list //查看所有环境
conda remove -n your_env_name --all //删除环境
conda clean --all //清理所有 Conda 缓存
conda create --name instantmesh python=3.10
conda activate instantmesh
pip install -U pip
// Ensure Ninja is installed
conda install Ninja
// Install the correct version of CUDA
conda install cuda -c nvidia/label/cuda-12.1.0
// Install PyTorch and xformers
// You may need to install another xformers version if you use a different PyTorch version
pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
pip install xformers==0.0.22.post7
// Install Triton
pip install triton
// Install other requirements
pip install -r requirements.txt
二、安装Pytorch GPU版
查看服务器的CUDA版本: nvidia-smi:
进入 pytorch官网
再安装
pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 --index-url https://download.pytorch.org/whl/cu118
安装完成后,测试。
~$ python
>>>import torch
>>>torch.cuda.is_available()
True
安装其他包
pip install -r ./3D/Unique3D-main/requirements-detail.txt
三、多卡训练
运行python脚本的参数
将
python 3D/Unique3D_main/app/gradio_local.py --port 7860
改成
python -m torch.distributed.launch --nproc_per_node=8 3D/Unique3D_main/app/gradio_local.py --port 7860