Commit 815bdd05 authored by Eric Hambro's avatar Eric Hambro
Browse files

Fixup to reflect Heiner's changes to TB.

parent bade7570
# -*- mode: dockerfile -*-
FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04
ARG PYTHON_VERSION=3.8
ARG PYTHON_VERSION=3.7
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -yq \
......@@ -43,7 +43,7 @@ RUN python -c 'import torch'
# Install TorchBeast
WORKDIR /opt/
RUN git clone https://github.com/condnsdmatters/torchbeast.git --branch eric/experimental-port --recursive
RUN git clone https://github.com/facebookresearch/torchbeast.git --recursive
WORKDIR /opt/torchbeast
RUN pip install -r requirements.txt
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment