added eigen requriement and installation

This commit is contained in:
Guocheng Qian
2023-08-08 18:50:42 +00:00
parent 4ad512ad41
commit a99ac494bc
7 changed files with 12 additions and 178 deletions

View File

@@ -6,13 +6,13 @@ module load eigen
# for aws ubuntu. install eigen
#sudo apt update && sudo apt upgrade
#sudo apt install libeigen3-dev
#sudo apt install git wget libeigen3-dev -y
# 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
python3 -m venv venv_magic123
source venv_magic123/bin/activate
# use conda
@@ -21,4 +21,4 @@ source venv_magic123/bin/activate
pip3 install torch torchvision
pip3 install -r requirements.txt
bash scripts/install_ext.sh
bash scripts/install_ext.sh