安装apex

来碗原味的小米粥吧

于 2024-05-22 19:27:22 发布

阅读量278

点赞数 3

文章标签： python

本文链接：https://blog.csdn.net/weixin_46046227/article/details/139126753

版权

虚拟环境

# 创建虚拟环境
conda create -n env_apex python=3.8.0
# 进入虚拟环境
conda activate env_apex

清空镜像

conda config --remove-key channels

安装pytorch

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

pytorch官网，生成带有cuda的安装pytorch的指令

更新环境变量

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
source ~/.bashrc

测试是否安装成功，返回true即为成功

import torch
print("installing_apex.py")
print(torch.version.cuda)
print(torch.__version__)
print(torch.cuda.is_available())

使用清华源镜像安装opencv最新版

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python

安装apex

# 我得到报错连接不进去，所以是下载到本地又解压的
# git clone https://github.com/NVIDIA/apex.git
cd apex
# 如果conda虚拟环境在切换目录后python路径出问题要用虚拟环境python的绝对路径
#如 /home/test/anaconda/envs/bwh/bin/python3.8 setup.py install
python setup.py install --cpp_ext --cuda_ext

在安装apex之前，先按照开源代码的readme按照yaml文件安装了一些包

 conda env update -f env_apex_pip.yaml

env_apex_pip.yaml内容

name: env_apex
dependencies:
  - pip:
    - easydict==1.9
    - opencv-python==4.5.4.58
    - pbkdf2==1.3
    - protobuf==3.19.4
    - scipy==1.7.3
    - tensorboardx==2.6
    - torchsummary==1.5.1
    - torchvision==0.15.2
prefix: /anaconda/envs/env_apex
# pytorch版本=2.1.0
# torchvision版本=0.16.0
# cuda版本=cuda11.8/12.1
# python版本=[3.8,3.11]