Kiss3DGen / pipeline /run_hpc.sh
JiantaoLin
new
02a9751
raw
history blame
419 Bytes
source /hpc2ssd/softwares/anaconda3/bin/activate kiss3dgen
module load cuda/12.1 compilers/gcc-11.1.0 compilers/icc-2023.1.0 cmake/3.27.0
export CXX=$(which g++)
export CC=$(which gcc)
export CPLUS_INCLUDE_PATH=/hpc2ssd/softwares/cuda/cuda-12.1/targets/x86_64-linux/include:$CPLUS_INCLUDE_PATH
export CUDA_LAUNCH_BLOCKING=1
export NCCL_TIMEOUT=3600
export CUDA_VISIBLE_DEVICES="0,1"
python ./pipeline/kiss3d_wrapper.py