pyton=3.8
cudn=11.7
附:查看cuda版本:nvcc -V
- 使用anaconda安装python3.8环境
- 进入python环境
conda activate py38
- 根据对应版本安装torch。在该链接中寻找对应版本Start Locally | PyTorch
conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia
- 安装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" ./
- 安装tqdm等库
pip install tqdm pip install scipy pip install -U scikit-learn