Skip to content

Commit

Permalink
minor
Browse files Browse the repository at this point in the history
  • Loading branch information
aghaeifar committed Jul 31, 2024
1 parent 0cae957 commit ec6ad9a
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 20 deletions.
4 changes: 2 additions & 2 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -8,9 +8,9 @@ find_package(HDF5 REQUIRED COMPONENTS CXX)
# Find OpenMP
find_package(OpenMP REQUIRED)
# Boost
find_package(Boost COMPONENTS program_options log log_setup REQUIRED)
find_package(Boost CONFIG COMPONENTS program_options log log_setup REQUIRED)
# FindCUDA module
find_package(CUDA)
# find_package(CUDA)
# find_package(CUDAToolkit)
# Find TBB
find_package(TBB REQUIRED)
Expand Down
27 changes: 9 additions & 18 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -1,7 +1,5 @@
# FROM ubuntu:22.04
# FROM nvidia/cuda:12.0.1-runtime-ubuntu22.04
FROM nvidia/cuda:12.0.1-devel-ubuntu22.04

FROM nvidia/cuda:12.5.1-devel-ubuntu22.04
USER root

# needed for add-apt-repository
Expand All @@ -16,9 +14,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
libtbb-dev \
gcc-11 \
g++-11 \
python3.10 \
python3-pip \
python-is-python3 \
htop \
tmux \
net-tools \
Expand All @@ -27,7 +22,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
git \
libboost-all-dev \
libhdf5-dev \
python3-venv \
# python3.10 \
# python3-pip \
# python-is-python3 \
# python3-venv \
&& apt-get clean && apt-get -y autoremove && rm -rf /var/lib/apt/lists/*

# RUN apt-get install nvidia-cuda-toolkit
Expand All @@ -38,23 +36,16 @@ RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 100 \

RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 11 && update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-11 11

RUN apt-get clean \
&& apt-get -y autoremove \
&& rm -rf /var/lib/apt/lists/*
# RUN pip install --no-cache-dir --upgrade pip && \
# pip install --no-cache-dir numpy torch scipy nibabel ipython jupyter matplotlib tqdm h5py scikit-image scikit-learn seaborn virtualenv && rm -rf /root/.cache

RUN pip install --no-cache-dir --upgrade pip && \
pip install --no-cache-dir numpy torch scipy nibabel ipython jupyter matplotlib tqdm h5py pandas scikit-image scikit-learn seaborn virtualenv && rm -rf /root/.cache

# Clone the Git repository
COPY . /opt/SpinWalk/
# Clone the Git repository
# RUN git clone --depth 1 https://github.com/aghaeifar/SpinWalk.git /opt/SpinWalk
WORKDIR /opt/SpinWalk
RUN cmake -B ./build && cmake --build ./build --config Release
RUN cmake --install ./build

RUN cmake -B ./build && cmake --build ./build --config Release && cmake --install ./build

LABEL org.opencontainers.image.authors="Ali Aghaeifar"


# docker build -t spinwalk .
# docker run --gpus all --rm -it --runtime=nvidia spinwalk bash

0 comments on commit ec6ad9a

Please sign in to comment.