Skip to content

Commit

Permalink
Merge pull request #48 from getwilds/version-specification-rtorch
Browse files Browse the repository at this point in the history
Specified versions in Dockerfile
  • Loading branch information
tefirman authored Oct 24, 2024
2 parents 36df84e + 8888a2b commit b3cb522
Show file tree
Hide file tree
Showing 2 changed files with 24 additions and 2 deletions.
22 changes: 22 additions & 0 deletions rtorch/Dockerfile_0.13.0
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@


# Using the Nvidia base image
# FROM rocker/tidyverse:3.6.0
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04

# Adding labels for the GitHub Container Registry
LABEL org.opencontainers.image.title="rtorch"
LABEL org.opencontainers.image.description="Container image for the use of the Torch R package in FH DaSL's WILDS"
LABEL org.opencontainers.image.version="0.13.0"
LABEL org.opencontainers.image.authors="[email protected]"
LABEL org.opencontainers.image.url=https://hutchdatascience.org/
LABEL org.opencontainers.image.documentation=https://getwilds.org/
LABEL org.opencontainers.image.source=https://github.com/getwilds/wilds-docker-library
LABEL org.opencontainers.image.licenses=MIT

# Installing Torch
ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update \
&& apt-get install -y --no-install-recommends build-essential=12.9ubuntu3 r-base=4.1.2-1ubuntu2 \
&& rm -rf /var/lib/apt/lists/*
RUN R -e "install.packages('remotes', repos = 'https://cloud.r-project.org'); remotes::install_version('torch', version = '0.13.0', repos = 'https://cloud.r-project.org'); torch::install_torch()"
4 changes: 2 additions & 2 deletions rtorch/Dockerfile_latest
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@


# Using the Ubuntu base image
# Using the Nvidia base image
# FROM rocker/tidyverse:3.6.0
FROM nvidia/cuda:11.7.1-cudnn8-devel-ubuntu22.04

Expand All @@ -19,4 +19,4 @@ ENV DEBIAN_FRONTEND noninteractive
RUN apt-get update \
&& apt-get install -y --no-install-recommends build-essential=12.9ubuntu3 r-base=4.1.2-1ubuntu2 \
&& rm -rf /var/lib/apt/lists/*
RUN R -e "install.packages('torch'); torch::install_torch()"
RUN R -e "install.packages('remotes', repos = 'https://cloud.r-project.org'); remotes::install_version('torch', version = '0.13.0', repos = 'https://cloud.r-project.org'); torch::install_torch()"

0 comments on commit b3cb522

Please sign in to comment.