Loading

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)可能出现如下报错:

截屏2023-02-09 23.49.12

解决方法:首先:

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

posted @ 2023-02-11 13:55  脂环  阅读(214)  评论(0编辑  收藏  举报