2021.7.30
训练环境:
Python 3.6.5
CUDA 10.0
Pytorch 1.2.0
Cudnn 7.6..2
Torchvision 0.4
Cudatoolkit 10.0.130
Ubutun16.04.6
GCC 5.4.0
Maskrcnn训练进程:
新建虚拟环境
Conda create -n maskrcnn2 python=3.6.5
- 安装合适的pip以及依赖
conda install ipython pip
- 安装cocoapi
pip install ninja yacs cython matplotlib tqdm opencv-python
可能是pip版本问题,先更新pip
Pip install –upgrade pip
Pip 更新成功,安装cocoqpi,更新成功
- 安装pytorch
conda install -c pytorch pytorch-nightly torchvision cudatoolkit=9.0
报错:
更换命令:pip3 install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html