Skip to content

Commit cd715ce

Browse files
committed
Add nvcc to all bases
1 parent c7986df commit cd715ce

File tree

3 files changed

+9
-4
lines changed

3 files changed

+9
-4
lines changed

build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/common.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,7 +55,7 @@ $APT_INSTALL \
5555
openssh-server \
5656
pkg-config \
5757
psmisc \
58-
python3 \
58+
python3-full \
5959
python3-pip \
6060
python3-venv \
6161
rar \
@@ -88,8 +88,8 @@ apt update
8888
locale-gen en_US.UTF-8
8989

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

9595
# Get Cloudflare daemon

build/COPY_ROOT_0/opt/ai-dock/bin/build/layer0/nvidia.sh

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,3 +10,8 @@ env-store CUDA_VERSION
1010
export CUDA_LEVEL=$(printf "%s" "$CUDA_STRING" | cut -d'-' -f2)
1111
env-store CUDA_LEVEL
1212

13+
# Ensure nvcc available on all bases
14+
cuda_version_dash=$(echo "$CUDA_VERSION" | sed -E 's/\.[^.]*$//; s/\./-/g')
15+
if [[ ! $CUDA_LEVEL == *devel* ]]; then
16+
$APT_INSTALL "cuda-nvcc-$cuda_version_dash"
17+
fi

build/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,7 +51,7 @@ ENV USER_NAME=user
5151
ENV USER_PASSWORD=password
5252
ENV USER_GROUPS=users,ai-dock,adm,sudo,audio,video,tty,cdrom,dialout,dip,fax,floppy,lp,plugdev,ssl-cert,sudo,tape,voice
5353
ENV DEBIAN_FRONTEND=noninteractive
54-
ENV PATH=/opt/ai-dock/bin:/opt/micromamba/bin:/opt/caddy/bin:$PATH
54+
ENV PATH=/opt/ai-dock/bin:/opt/caddy/bin:$PATH
5555
ENV OPT_SYNC=
5656
ENV PYTHONUNBUFFERED=true
5757
ENV VENV_DIR=/opt/environments/python

0 commit comments

Comments
 (0)