场景分割环境搭建 pyton==3.8 cuda==11.7 pytorch

pyton=3.8

cudn=11.7

附：查看cuda版本：nvcc -V

1. 使用anaconda安装python3.8环境
2. 进入python环境

   conda activate py38
   

3. 根据对应版本安装torch。在该链接中寻找对应版本Start Locally | PyTorch

   conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
   

4. 安装apex。从github上下载资源Apex (A PyTorch Extension) — Apex 0.1.0 documentation

   cd apex/
   pip install packaging
   
   vim setup.py #修改setup.py的cuda_dir
   # cuda_dir = "/usr/local/cuda-11.7"
   
   pip install -v --no-cache-dir ./
   #pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
   
   

5. 安装tqdm等库

   pip install tqdm
   pip install scipy
   pip install -U scikit-learn