File tree Expand file tree Collapse file tree 4 files changed +13
-10
lines changed Expand file tree Collapse file tree 4 files changed +13
-10
lines changed Original file line number Diff line number Diff line change @@ -7,7 +7,7 @@ opencv-python>=4.1.1
7
7
Pillow >= 7.1.2
8
8
PyYAML >= 5.3.1
9
9
requests >= 2.23.0
10
- scipy >= 1.5
10
+ scipy >= 1.4.1 # Google Colab version
11
11
torch >= 1.7.0
12
12
torchvision >= 0.8.1
13
13
tqdm >= 4.41.0
Original file line number Diff line number Diff line change @@ -6,13 +6,12 @@ FROM nvcr.io/nvidia/pytorch:21.10-py3
6
6
# Install linux packages
7
7
RUN apt update && apt install -y zip htop screen libgl1-mesa-glx
8
8
9
- # Install python dependencies
9
+ # Install pip packages
10
10
COPY requirements.txt .
11
11
RUN python -m pip install --upgrade pip
12
12
RUN pip uninstall -y torch torchvision torchtext
13
13
RUN pip install --no-cache -r requirements.txt albumentations wandb gsutil notebook \
14
14
torch==1.11.0+cu113 torchvision==0.12.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
15
- # RUN pip install --no-cache -U torch torchvision
16
15
17
16
# Create working directory
18
17
RUN mkdir -p /usr/src/app
Original file line number Diff line number Diff line change @@ -6,15 +6,19 @@ FROM ubuntu:latest
6
6
# Install linux packages
7
7
RUN apt update
8
8
RUN DEBIAN_FRONTEND=noninteractive TZ=Etc/UTC apt install -y tzdata
9
- RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0
10
- RUN alias python=python3
9
+ RUN apt install -y python3-pip git zip curl htop screen libgl1-mesa-glx libglib2.0-0 software-properties-common
11
10
12
- # Install python dependencies
11
+ # Install python3.9
12
+ RUN add-apt-repository ppa:deadsnakes/ppa -y
13
+ RUN apt install python3.9 python3.9-distutils libpython3.9 -y
14
+ # RUN alias python=python3.9
15
+
16
+ # Install pip packages
13
17
COPY requirements.txt .
14
- RUN python3 -m pip install --upgrade pip
15
- RUN pip install --no-cache -r requirements.txt albumentations gsutil notebook \
18
+ RUN python3.9 -m pip install --upgrade pip
19
+ RUN python3.9 -m pip install --no-cache -r requirements.txt albumentations gsutil notebook \
16
20
coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu tensorflowjs \
17
- torch==1.11.0+cpu torchvision==0.12.0+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html
21
+ torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
18
22
19
23
# Create working directory
20
24
RUN mkdir -p /usr/src/app
Original file line number Diff line number Diff line change @@ -467,7 +467,7 @@ def check_dataset(data, autodownload=True):
467
467
if val :
468
468
val = [Path (x ).resolve () for x in (val if isinstance (val , list ) else [val ])] # val path
469
469
if not all (x .exists () for x in val ):
470
- LOGGER .info (emojis ('\n Dataset not found ⚠️ , missing paths %s' % [str (x ) for x in val if not x .exists ()]))
470
+ LOGGER .info (emojis ('\n Dataset not found ⚠, missing paths %s' % [str (x ) for x in val if not x .exists ()]))
471
471
if s and autodownload : # download script
472
472
t = time .time ()
473
473
root = path .parent if 'path' in data else '..' # unzip directory i.e. '../'
You can’t perform that action at this time.
0 commit comments