Skip to content
Snippets Groups Projects
Commit bcbabd90 authored by Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)'s avatar Dutta, Ayushi (PG/R - Comp Sci & Elec Eng)
Browse files

cuda compute added

parent f70c6109
No related branches found
No related tags found
No related merge requests found
Pipeline #102066 failed
......@@ -14,11 +14,14 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh
bash miniconda.sh -b -p /opt/conda
rm miniconda.sh
# Code environment setup
export PATH=/opt/conda/bin:$PATH
export TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6+PTX"
export TCNN_CUDA_ARCHITECTURES="75;80;86"
. /opt/conda/etc/profile.d/conda.sh # Non - interactive sessions we need this
conda init bash # For future interactive sessions, optional
# Conda environment
git clone https://github.com/snuvclab/gala.git
cd gala
conda env create -f env.yaml
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment