Forked from stonehye/gist:56de90d54a915e23b3202a77bebe2c56
Created
July 7, 2025 15:20
-
-
Save anjilinux/62f2ff37613e85c41400561e1aa84e16 to your computer and use it in GitHub Desktop.
Dockerfile - pytorch-ubuntu16.04 with only pip (without conda)
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04 | |
| RUN apt-get update && apt-get -y upgrade | |
| RUN apt-get install -y --no-install-recommends git wget | |
| RUN apt-get install -y software-properties-common vim ssh | |
| RUN apt-get install -y python3 | |
| RUN apt-get install -y python3-dev python3-pip build-essential | |
| RUN apt-get install -y software-properties-common vim ssh | |
| RUN pip3 install --upgrade pip | |
| RUN ln -s /usr/bin/python3 /usr/bin/python | |
| RUN pip install torch torchvision | |
| WORKDIR /workspace | |
| ADD . . | |
| RUN chmod -R a+w /workspace | |
| EXPOSE 8000 | |
| EXPOSE 7822 |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment