Python服务器创建虚拟环境跑代码

conda create -n KDDocRE python==3.7.4

conda activate KDDocRE

conda install cudatoolkit=10.2

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge 
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/    
conda config --set show_channel_urls yes

pip3 install torch torchvision torchaudio 

或者

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

pip install transformers==4.8.2

pip install numpy==1.19.4

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" ./

pip install opt_einsum==3.3.0

pip install axial_attention==0.6.1

pip install ujson

pip install tqdm

conda deactivate

conda env list

conda remove -n KDDocRE