Skip to content

Commit ed48220

Browse files
committed
update
1 parent 43e56e8 commit ed48220

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

.devcontainer/Dockerfile

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,16 +1,11 @@
1-
FROM mcr.microsoft.com/devcontainers/python:1-3.12-bullseye
1+
FROM --platform=linux/amd64 mcr.microsoft.com/devcontainers/python:1-3.12-bullseye
22

33
RUN --mount=type=cache,target=/home/vscode/.cache/pip \
44
set -eux; \
55
apt-get update; \
66
apt-get install -y moreutils; \
77
pip wheel --no-deps torch; \
8-
pip install patchelf torch-*.whl
9-
10-
# link all /usr/local/lib/python3.12/site-packages/nvidia/*/lib
11-
# to /etc/ld.so.conf.d/nv.conf
12-
# and run ldconfig
13-
RUN set -eux; \
8+
pip install patchelf torch-*.whl; \
149
mkdir -p /usr/local/lib/nv; \
1510
ln -s /usr/local/lib/python3.12/site-packages/nvidia/*/lib/*.so* /usr/local/lib/nv/; \
1611
echo "/usr/local/lib/nv" > /etc/ld.so.conf.d/nv.conf; \

.devcontainer/demo.sh

100644100755
Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,11 @@
1+
#! /bin/bash
2+
3+
pip install -e /workspaces/auditwheel
4+
15
auditwheel -v repair \
26
--exclude libcuds.so.1 \
37
--exclude libcusolver.so.11 \
48
--exclude libcusparseLt.so.0 \
59
--plat=manylinux_2_35_x86_64 \
6-
/torch-2.6.0-cp312-cp312-manylinux1_x86_64.whl
10+
/torch-2.6.0-cp312-cp312-manylinux1_x86_64.whl \
11+
2>&1 | ts '[%Y-%m-%d %H:%M:%S]'

0 commit comments

Comments
 (0)