CVPR2021 PLOP 论文+CLIP代码环境搭建
查看cuda版本(10.1):
nvcc -V
创建python3.7虚拟环境:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.1 -c pytorch
如果conda出问题可以试试pip(https://pytorch.org/get-started/previous-versions/):
pip install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
这条命令对应的cuda版本是10.2,然而1.8.0的torch在使用CLIP的时候可能会报错(参考https://www.cnblogs.com/lipoicyclic/p/17438507.html以及https://www.cnblogs.com/lipoicyclic/p/17481304.html),最好装一个和cuda版本对应的、大于等于1.8.1的torch:
pip install torch==1.8.1+cu102 torchvision==0.9.1+cu102 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
安装apex:
conda install packaging
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
如果pytorch版本比较老(比如1.8.0)可能出现如下报错:
解决方法:首先:
pip uninstall apex
然后按照图中所说:
git clone -b 22.04-dev https://github.com/NVIDIA/apex.git
安装apex可能会报错gcc版本大于8,参考https://zhuanlan.zhihu.com/p/453542931降低gcc和g++的版本。
安装inplace_abn:
从源码安装会报setuptools_scm找不到版本的一个错,直接:
pip install inplace-abn
安装matplotlib:
pip install matplotlib
matplotlib可能会导致报错ImportError lib64libstdc++.so.6 version `GLIBCXX_3.4.29’ not found,可以参考https://blog.csdn.net/zyczjx/article/details/126904670也可以直接把stream_metrics.py的相关部分注释掉。
安装clip:
服务器没代理,可以本地挂代理把源码下下来,传到服务器上,进入文件夹安装:
python setup.py build install
安装tensorboardX:
pip install tensorboardX
注,clone不下来可以挂代理:
export ALL_PROXY=socks5://127.0.0.1:7890
如果代理寄了就重置。
之后安装:
conda install typing_extensions
以及conda install IPython