diff options
| author | Amir Arsalan Soltani <Amir-Arsalan@users.noreply.github.com> | 2018-02-12 14:43:28 -0500 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-02-12 14:43:28 -0500 |
| commit | 2fd8e596b62205cb4ab395f458b5d81e0963ddc2 (patch) | |
| tree | aa9bdaaa9d8e28301b74604abdd787044651c1c3 /Dockerfile | |
| parent | 4ed87e3c9e110a8af6b8e4186ee480edf78e077c (diff) | |
| download | pytorch-2fd8e596b62205cb4ab395f458b5d81e0963ddc2.tar.gz pytorch-2fd8e596b62205cb4ab395f458b5d81e0963ddc2.tar.bz2 pytorch-2fd8e596b62205cb4ab395f458b5d81e0963ddc2.zip | |
CUDA 9 (#5194)
Diffstat (limited to 'Dockerfile')
| -rw-r--r-- | Dockerfile | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Dockerfile b/Dockerfile index 0ea551e1f3..c5b1bbaee6 100644 --- a/Dockerfile +++ b/Dockerfile @@ -25,7 +25,7 @@ RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-la /opt/conda/bin/conda create -y --name pytorch-py$PYTHON_VERSION python=$PYTHON_VERSION numpy pyyaml scipy ipython mkl&& \ /opt/conda/bin/conda clean -ya ENV PATH /opt/conda/envs/pytorch-py$PYTHON_VERSION/bin:$PATH -RUN conda install --name pytorch-py$PYTHON_VERSION -c soumith magma-cuda80 +RUN conda install --name pytorch-py$PYTHON_VERSION -c soumith magma-cuda90 # This must be done before pip so that requirements.txt is available WORKDIR /opt/pytorch COPY . . |