remove thop dependency
This commit is contained in:
parent
ee8988b8a2
commit
11b462e2a9
|
|
@ -2,7 +2,7 @@
|
||||||
FROM nvcr.io/nvidia/pytorch:20.03-py3
|
FROM nvcr.io/nvidia/pytorch:20.03-py3
|
||||||
|
|
||||||
# Install dependencies (pip or conda)
|
# Install dependencies (pip or conda)
|
||||||
RUN pip install -U gsutil thop
|
RUN pip install -U gsutil
|
||||||
# RUN pip install -U -r requirements.txt
|
# RUN pip install -U -r requirements.txt
|
||||||
|
|
||||||
# Create working directory
|
# Create working directory
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue