Skip to content

Commit

Permalink
fix(deploy): Enable cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
jourdain committed Jul 12, 2023
1 parent fdc35ae commit 7ca6959
Show file tree
Hide file tree
Showing 3 changed files with 67 additions and 1 deletion.
3 changes: 3 additions & 0 deletions .deploy-release/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,3 +1,6 @@
FROM kitware/trame:glvnd
COPY --chown=trame-user:trame-user .deploy-release/setup /deploy/setup
RUN /opt/trame/entrypoint.sh build

ENV NVIDIA_DRIVER_CAPABILITIES=graphics,utility,compute
ENV TRAME_WS_HEART_BEAT=300
1 change: 1 addition & 0 deletions .deploy-release/setup/initialize.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
64 changes: 63 additions & 1 deletion .deploy-release/setup/requirements.txt
Original file line number Diff line number Diff line change
@@ -1 +1,63 @@
xaitk-saliency-demo
aiohttp==3.8.4
aiosignal==1.3.1
async-timeout==4.0.2
attrs==23.1.0
certifi==2023.5.7
charset-normalizer==3.2.0
click==8.1.4
frozenlist==1.3.3
idna==3.4
imageio==2.31.1
joblib==1.3.1
lazy_loader==0.3
llvmlite==0.40.1
multidict==6.0.4
networkx==3.1
numba==0.57.1
numpy==1.24.4
opencv-python==4.8.0.74
packaging==23.1
pandas==2.0.3
Pillow==10.0.0
pkg_resources==0.0.0
plotly==5.15.0
python-dateutil==2.8.2
pytz==2023.3
PyWavelets==1.4.1
PyYAML==6.0
requests==2.31.0
scikit-image==0.21.0
scikit-learn==1.3.0
scipy==1.11.1
six==1.16.0
smqtk-classifier==0.19.0
smqtk-core==0.19.0
smqtk-dataprovider==0.18.0
smqtk-descriptors==0.19.0
smqtk-detection==0.20.0
smqtk-image-io==0.17.1
tenacity==8.2.2
threadpoolctl==3.1.0
tifffile==2023.7.10
trame==2.5.0
trame-client==2.9.4
trame-components==2.1.1
trame-deckgl==2.0.2
trame-markdown==2.0.2
trame-matplotlib==2.0.2
trame-plotly==2.1.1
trame-rca==0.3.1
trame-router==2.0.2
trame-server==2.11.4
trame-simput==2.3.2
trame-vega==2.0.3
trame-vtk==2.5.4
trame-vuetify==2.2.4
typing_extensions==4.7.1
tzdata==2023.3
ubelt==1.3.2
urllib3==2.0.3
wslink==1.11.1
xaitk-saliency==0.7.0
xaitk-saliency-demo==2.2.0
yarl==1.9.2

0 comments on commit 7ca6959

Please sign in to comment.