File size: 423 Bytes
4157d39 02a9751 |
1 2 3 4 5 6 7 8 9 10 |
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/example_text_to_3d.py |