update

Author: oraluben

.devcontainer/Dockerfile

@@ -1,16 +1,11 @@
-FROM mcr.microsoft.com/devcontainers/python:1-3.12-bullseye
+FROM --platform=linux/amd64 mcr.microsoft.com/devcontainers/python:1-3.12-bullseye
 
 RUN --mount=type=cache,target=/home/vscode/.cache/pip \
     set -eux; \
     apt-get update; \
     apt-get install -y moreutils; \
     pip wheel --no-deps torch; \
-    pip install patchelf torch-*.whl
-
-# link all /usr/local/lib/python3.12/site-packages/nvidia/*/lib
-# to /etc/ld.so.conf.d/nv.conf
-# and run ldconfig
-RUN set -eux; \
+    pip install patchelf torch-*.whl; \
     mkdir -p /usr/local/lib/nv; \
     ln -s /usr/local/lib/python3.12/site-packages/nvidia/*/lib/*.so* /usr/local/lib/nv/; \
     echo "/usr/local/lib/nv" > /etc/ld.so.conf.d/nv.conf; \

.devcontainer/demo.sh

@@ -1,6 +1,10 @@
+#! /bin/bash
+
 auditwheel -v repair \
     --exclude libcuds.so.1 \
     --exclude libcusolver.so.11 \
     --exclude libcusparseLt.so.0 \
     --plat=manylinux_2_35_x86_64 \
-    /torch-2.6.0-cp312-cp312-manylinux1_x86_64.whl
+    -w /tmp/wheelhouse \
+    /torch-2.6.0-cp312-cp312-manylinux1_x86_64.whl \
+    2>&1 | ts '[%Y-%m-%d %H:%M:%S]'

.devcontainer/devcontainer.json

@@ -3,5 +3,6 @@
 	"mounts": [
     	"type=bind,source=${localEnv:HOME}${localEnv:USERPROFILE}/.ssh/id_rsa,target=/home/vscode/.ssh/id_rsa,readonly",
     	"type=bind,source=${localEnv:HOME}${localEnv:USERPROFILE}/.ssh/known_hosts,target=/home/vscode/.ssh/known_hosts"
-	]
+	],
+	"postStartCommand": "pip install -e /workspaces/auditwheel"
 }