**编译安装步骤**


- 1. 安装与dtk版本对应的pytorch torchvision whl（可在光合社区下载https://developer.hpccube.com/tool/）以及其他依赖库
    

    例如：
  
    pip install https://cancon.hpccube.com:65024/directlink/4/pytorch/dtk22.10/torch-1.10.0a0+git2040069.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
  
    pip install https://cancon.hpccube.com:65024/directlink/4/vision/dtk22.10/torchvision-0.10.0a0+e04d001.dtk2210-cp38-cp38-manylinux2014_x86_64.whl
    
    pip install -r requirements.txt

- 2. 添加编译时的conda环境及部分库的环境变量
    
    - 2.1 激活对应的conda环境：

 
      source ~/miniconda3/etc/profile.d/conda.sh

      conda activate torch1.10_py39_dtk22.10

    - 2.2 加载对应的module，包括dtk：

 
      module purge

      module load compiler/devtoolset/7.3.1 mpi/hpcx/gcc-7.3.1 compiler/dtk/22.10.1

      module list

    - 2.3 加载所需的依赖库的环境变量（根据各集群实际路径调整）：

 
      export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$C_INCLUDE_PATH

      export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/gflags-2.1.2-build/include:$CPLUS_INCLUDE_PATH

      export C_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$C_INCLUDE_PATH

      export CPLUS_INCLUDE_PATH=/public/software/apps/DeepLearning/PyTorch_Lib/glog-build/include:$CPLUS_INCLUDE_PATH

      export C_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$C_INCLUDE_PATH

      export CPLUS_INCLUDE_PATH=$ROCM_PATH/rocrand/include:$CPLUS_INCLUDE_PATH

      export LD_LIBRARY_PATH=$ROCM_PATH/rocrand/lib:$LD_LIBRARY_PATH

    - 2.4 修改编译器环境变量：
  

      export FORCE_ONLY_HIP=1

      export CC=hipcc

      export CXX=hipcc

- 3. 编译安装
  

      python setup.py install