美文网首页
docker mmdet 更新(内部)

docker mmdet 更新(内部)

作者: 默写年华Antifragile | 来源:发表于2020-02-16 00:01 被阅读0次

修改清华源

vi /root/.condarc
channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud

升级pytorch(1.4 cuda10.1)

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

重新编译 mmdet

cd /mmdetection
rm -r build/
pip install -v -e .

相关文章

网友评论

      本文标题:docker mmdet 更新(内部)

      本文链接:https://www.haomeiwen.com/subject/itbufhtx.html