Skip to content

Commit a71cec7

Browse files
committed
fix: update train entrypoint script to allow insecure SSL connections
1 parent ee97dc4 commit a71cec7

2 files changed

Lines changed: 2 additions & 6 deletions

File tree

docker/train_entrypoint.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ if [[ -n "${_DRIVER_VERSION:-}" ]]; then
3535
mkdir -p "${_NVIDIA_CACHE}"
3636
_DRIVER_URL="https://us.download.nvidia.com/tesla/${_DRIVER_VERSION}/NVIDIA-Linux-x86_64-${_DRIVER_VERSION}.run"
3737
_DRIVER_RUN="/tmp/nvidia-${_DRIVER_VERSION}.run"
38-
if curl -fsSL "${_DRIVER_URL}" -o "${_DRIVER_RUN}"; then
38+
if curl -fsSLk "${_DRIVER_URL}" -o "${_DRIVER_RUN}"; then
3939
chmod +x "${_DRIVER_RUN}"
4040
"${_DRIVER_RUN}" --extract-only --target "${_NVIDIA_CACHE}-extract"
4141
cp "${_NVIDIA_CACHE}-extract"/*.so.* "${_NVIDIA_CACHE}/" 2>/dev/null || true

nix/docker.nix

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -126,14 +126,13 @@ let
126126
rustPackages."psyche-sidecar"
127127
python3Packages.huggingface-hub
128128
(pkgs.runCommand "entrypoint" { } ''
129-
mkdir -p $out/bin $out/etc/ssl/certs $out/tmp $out/var/tmp $out/run
129+
mkdir -p $out/bin $out/etc $out/tmp $out/var/tmp $out/run
130130
cp ${../docker/train_entrypoint.sh} $out/bin/train_entrypoint.sh
131131
cp ${../docker/sidecar_entrypoint.sh} $out/bin/sidecar_entrypoint.sh
132132
chmod +x $out/bin/train_entrypoint.sh
133133
chmod +x $out/bin/sidecar_entrypoint.sh
134134
ln -s ${bashInteractive}/bin/bash $out/bin/bash
135135
ln -s ${bashInteractive}/bin/bash $out/bin/sh
136-
ln -s ${cacert}/etc/ssl/certs/ca-certificates.crt $out/etc/ssl/certs/ca-certificates.crt
137136
'')
138137
]
139138
++ cudaRuntimeLibs;
@@ -144,9 +143,6 @@ let
144143
"NVIDIA_VISIBLE_DEVICES=all"
145144
"LD_LIBRARY_PATH=${lib.makeLibraryPath cudaRuntimeLibs}:/usr/local/nvidia/lib64:/lib:/usr/lib"
146145
"PATH=${lib.makeBinPath (with pkgs; [ bashInteractive coreutils curl findutils gnugrep ])}"
147-
"SSL_CERT_FILE=${pkgs.cacert}/etc/ssl/certs/ca-certificates.crt"
148-
"CURL_CA_BUNDLE=${pkgs.cacert}/etc/ssl/certs/ca-certificates.crt"
149-
"NIX_SSL_CERT_FILE=${pkgs.cacert}/etc/ssl/certs/ca-certificates.crt"
150146
"LOGNAME=root"
151147
"TORCHINDUCTOR_CACHE_DIR=/tmp/torchinductor"
152148
"PYTHONUNBUFFERED=1"

0 commit comments

Comments
 (0)