首先要确保nvcc版本和要装的cuda版本一致,从而能够成功安装apex库,查看当前nvcc的版本:
nvcc --version
一开始我的版本是10.0,去官网上下载了10.2的更新
然后就可以开始配置环境了,首先创建虚拟环境
conda create -n SATS python=3.8 -y
conda activate SATS
然后安装pytorch,版本好像不能太高,不然apex库会安装失败,一开始我装的1.10.0版本pytorch就不行
conda install pytorch==1.8.1 torchvision cudatoolkit=10.2 -c pytorch
然后安装apex库
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
中间可能会报一些错,最后只要显示如下信息就说明安装成功了
接着只要安装一些剩余的依赖环境即可
pip install tqdm
pip install matplotlib
pip install einops
pip install timm
pip install imgviz
pip install opencv-python-headless
pip install tensorboardX