Skip to content

Commit

Permalink
Add nvcc to all bases
Browse files Browse the repository at this point in the history
  • Loading branch information
robballantyne committed Jun 16, 2024
1 parent c7986df commit cd715ce
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 4 deletions.
6 changes: 3 additions & 3 deletions build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/common.sh
Original file line number Diff line number Diff line change
Expand Up @@ -55,7 +55,7 @@ $APT_INSTALL \
openssh-server \
pkg-config \
psmisc \
python3 \
python3-full \
python3-pip \
python3-venv \
rar \
Expand Down Expand Up @@ -88,8 +88,8 @@ apt update
locale-gen en_US.UTF-8

# Install
python3.10 -m venv /opt/environments/python/serviceportal
/opt/environments/python/serviceportal/bin/pip install \
python3.10 -m venv "$SERVICEPORTAL_VENV"
"$SERVICEPORTAL_VENV_PIP" install \
--no-cache-dir -r /opt/ai-dock/fastapi/requirements.txt

# Get Cloudflare daemon
Expand Down
5 changes: 5 additions & 0 deletions build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/nvidia.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,3 +10,8 @@ env-store CUDA_VERSION
export CUDA_LEVEL=$(printf "%s" "$CUDA_STRING" | cut -d'-' -f2)
env-store CUDA_LEVEL

# Ensure nvcc available on all bases
cuda_version_dash=$(echo "$CUDA_VERSION" | sed -E 's/\.[^.]*$//; s/\./-/g')
if [[ ! $CUDA_LEVEL == *devel* ]]; then
$APT_INSTALL "cuda-nvcc-$cuda_version_dash"
fi
2 changes: 1 addition & 1 deletion build/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ ENV USER_NAME=user
ENV USER_PASSWORD=password
ENV USER_GROUPS=users,ai-dock,adm,sudo,audio,video,tty,cdrom,dialout,dip,fax,floppy,lp,plugdev,ssl-cert,sudo,tape,voice
ENV DEBIAN_FRONTEND=noninteractive
ENV PATH=/opt/ai-dock/bin:/opt/micromamba/bin:/opt/caddy/bin:$PATH
ENV PATH=/opt/ai-dock/bin:/opt/caddy/bin:$PATH
ENV OPT_SYNC=
ENV PYTHONUNBUFFERED=true
ENV VENV_DIR=/opt/environments/python
Expand Down

0 comments on commit cd715ce

Please sign in to comment.