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

runtime baase

parent bcbabd90
No related branches found
No related tags found
No related merge requests found
Pipeline #102087 failed
# Get the base image from nvidia/cuda Docker image.
BootStrap: docker
From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
From: nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
%post
# Install special software or additions
......@@ -20,8 +20,9 @@ From: nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
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
mkdir -p /workspace
cd /workspace
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