first commit

This commit is contained in:
Guocheng Qian
2023-08-02 19:51:43 -07:00
parent c2891c38cc
commit 13e18567fa
202 changed files with 43362 additions and 17 deletions

24
install.sh Normal file
View File

@@ -0,0 +1,24 @@
# for KAUST cluster
module load cuda/11.7.0
module load gcc/7.5.0
module load eigen
# for aws ubuntu. install eigen
#sudo apt update && sudo apt upgrade
#sudo apt install libeigen3-dev
# a100: 8.0; v100: 7.0; 2080ti: 7.5; titan xp: 6.1
export TORCH_CUDA_ARCH_LIST="7.0;7.5;8.0"
# use python venv
python -m venv venv_magic123
source venv_magic123/bin/activate
# use conda
# conda create -n magic123 python=3.10 ipython -y
# conda activate magic123
pip3 install torch torchvision
pip3 install -r requirements.txt
bash scripts/install_ext.sh