Open rohitdhote111 opened 4 months ago
I have done installations using the requirements.txt file which is provided with the repo. (p.s. installed libraries details are added in the end)
root@81f78440247d:/home/workspace/StyleGenForLabels/pixel2style2pixel# python scripts/train.py \
--dataset_type=plants_seg_to_real_uav \ --exp_dir=/FENtraining6 \ --workers=0 \ --batch_size=16 \ --test_batch_size=16 \ --test_workers=0 \ --val_interval=5000 \ --save_interval=10000 \ --image_interval=1000 \ --encoder_type=GradualStyleEncoder \ --label_nc=3 \ --input_nc=3 \ --output_size=512 \ --max_steps=250000 \ --stylegan_weights=/home/workspace/stylegan2_weights.pt \ --learning_rate=0.0001 {'batch_size': 16, 'board_interval': 50, 'checkpoint_path': None, 'dataset_type': 'plants_seg_to_real_uav', 'encoder_type': 'GradualStyleEncoder', 'exp_dir': '/FENtraining6', 'id_lambda': 0, 'image_interval': 1000, 'img_w_pairs_lambda': 1.0, 'input_nc': 3, 'l2_lambda': 1.0, 'l2_lambda_crop': 0, 'label_nc': 3, 'learn_in_w': False, 'learning_rate': 0.0001, 'lpips_lambda': 0.8, 'lpips_lambda_crop': 0, 'max_steps': 250000, 'moco_lambda': 0, 'optim_name': 'ranger', 'output_size': 512, 'resize_factors': None, 'save_interval': 10000, 'start_from_latent_avg': False, 'stylegan_weights': '/home/workspace/stylegan2_weights.pt', 'test_batch_size': 16, 'test_workers': 0, 'train_decoder': False, 'use_wandb': False, 'val_interval': 5000, 'w_norm_lambda': 0, 'workers': 0} Loading decoder weights from pretrained! Loading dataset for plants_seg_to_real_uav Number of training samples: 379 Number of test samples: 647 Traceback (most recent call last): File "scripts/train.py", line 32, in main() File "scripts/train.py", line 27, in main coach = Coach(opts) File "./training/coach.py", line 76, in init self.logger = SummaryWriter(log_dir=log_dir) File "/usr/local/lib/python3.6/dist-packages/torch/utils/tensorboard/writer.py", line 220, in init self._get_file_writer() File "/usr/local/lib/python3.6/dist-packages/torch/utils/tensorboard/writer.py", line 251, in _get_file_writer self.flush_secs, self.filename_suffix) File "/usr/local/lib/python3.6/dist-packages/torch/utils/tensorboard/writer.py", line 61, in init log_dir, max_queue, flush_secs, filename_suffix) File "/usr/local/lib/python3.6/dist-packages/tensorboard/summary/writer/event_file_writer.py", line 76, in init if not tf.io.gfile.exists(logdir): File "/usr/local/lib/python3.6/dist-packages/tensorboard/lazy.py", line 68, in getattr return getattr(load_once(self), attr_name) AttributeError: module 'tensorflow' has no attribute 'io'
root@81f78440247d:/home/workspace/StyleGenForLabels# pip install -r requirements.txt Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple Requirement already satisfied: absl-py==1.0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 1)) (1.0.0) Requirement already satisfied: cachetools==4.2.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 2)) (4.2.4) Requirement already satisfied: certifi==2020.4.5.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 3)) (2020.4.5.1) Requirement already satisfied: charset-normalizer==2.0.12 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 4)) (2.0.12) Requirement already satisfied: click==8.0.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 5)) (8.0.4) Requirement already satisfied: cycler==0.11.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 6)) (0.11.0) Requirement already satisfied: dataclasses==0.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 7)) (0.8) Requirement already satisfied: docker-pycreds==0.4.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 8)) (0.4.0) Requirement already satisfied: future==0.18.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 9)) (0.18.2) Requirement already satisfied: gitdb==4.0.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 10)) (4.0.9) Requirement already satisfied: GitPython==3.1.18 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 11)) (3.1.18) Requirement already satisfied: google-auth==1.35.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 12)) (1.35.0) Requirement already satisfied: google-auth-oauthlib==0.4.6 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 13)) (0.4.6) Requirement already satisfied: grpcio==1.44.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 14)) (1.44.0) Requirement already satisfied: idna==3.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 15)) (3.3) Requirement already satisfied: importlib-metadata==4.8.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 16)) (4.8.3) Requirement already satisfied: joblib==1.1.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 17)) (1.1.0) Requirement already satisfied: kiwisolver==1.3.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 18)) (1.3.1) Requirement already satisfied: lmdb==1.3.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 19)) (1.3.0) Requirement already satisfied: Markdown==3.3.6 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 20)) (3.3.6) Requirement already satisfied: matplotlib==3.2.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 21)) (3.2.1) Requirement already satisfied: ninja==1.11.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 22)) (1.11.1) Requirement already satisfied: numpy==1.18.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 23)) (1.18.4) Requirement already satisfied: oauthlib==3.2.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 24)) (3.2.0) Requirement already satisfied: opencv-python==4.2.0.34 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 25)) (4.2.0.34) Requirement already satisfied: pathtools==0.1.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 26)) (0.1.2) Requirement already satisfied: Pillow==7.1.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 27)) (7.1.2) Requirement already satisfied: promise==2.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 28)) (2.3) Requirement already satisfied: protobuf==3.19.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 29)) (3.19.4) Requirement already satisfied: psutil==5.9.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 30)) (5.9.1) Requirement already satisfied: pyasn1==0.4.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 31)) (0.4.8) Requirement already satisfied: pyasn1-modules==0.2.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 32)) (0.2.8) Requirement already satisfied: pyparsing==3.0.7 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 33)) (3.0.7) Requirement already satisfied: python-dateutil==2.8.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 34)) (2.8.2) Requirement already satisfied: PyYAML==6.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 35)) (6.0) Requirement already satisfied: requests==2.27.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 36)) (2.27.1) Requirement already satisfied: requests-oauthlib==1.3.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 37)) (1.3.1) Requirement already satisfied: rsa==4.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 38)) (4.8) Requirement already satisfied: scikit-learn==0.24.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 39)) (0.24.2) Requirement already satisfied: scipy==1.4.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 40)) (1.4.1) Requirement already satisfied: sentry-sdk==1.9.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 41)) (1.9.0) Requirement already satisfied: setproctitle==1.2.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 42)) (1.2.3) Requirement already satisfied: shortuuid==1.0.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 43)) (1.0.9) Requirement already satisfied: six==1.16.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 44)) (1.16.0) Requirement already satisfied: sklearn==0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 45)) (0.0) Requirement already satisfied: smmap==5.0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 46)) (5.0.0) Collecting tensorboard==2.2.1 Using cached https://pypi.tuna.tsinghua.edu.cn/packages/1d/fd/4f3ca1516cbb3713259ef229abd9314bba0077ef6070285dde0dd1ed21b2/tensorboard-2.2.1-py3-none-any.whl (3.0 MB) Requirement already satisfied: tensorboard-plugin-wit==1.8.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 48)) (1.8.1) Requirement already satisfied: threadpoolctl==3.1.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 49)) (3.1.0) Requirement already satisfied: tqdm==4.46.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 50)) (4.46.0) Requirement already satisfied: typing-extensions==4.1.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 51)) (4.1.1) Requirement already satisfied: urllib3==1.26.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 52)) (1.26.9) Requirement already satisfied: wandb==0.13.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 53)) (0.13.0) Requirement already satisfied: Werkzeug==2.0.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 54)) (2.0.3) Requirement already satisfied: zipp==3.6.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 55)) (3.6.0) Requirement already satisfied: scikit-image in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 56)) (0.17.2) Requirement already satisfied: torchmetrics in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 57)) (0.8.2) Requirement already satisfied: oyaml in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 58)) (1.0) Requirement already satisfied: pytorch_lightning in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 59)) (1.5.10) Requirement already satisfied: setuptools>=40.3.0 in /usr/local/lib/python3.6/dist-packages (from google-auth==1.35.0->-r requirements.txt (line 12)) (59.5.0) Requirement already satisfied: wheel>=0.26 in /usr/lib/python3/dist-packages (from tensorboard==2.2.1->-r requirements.txt (line 47)) (0.30.0) Requirement already satisfied: fsspec[http]!=2021.06.0,>=2021.05.0 in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (2022.1.0) Requirement already satisfied: torch>=1.7.* in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (1.8.1+cu111) Requirement already satisfied: packaging>=17.0 in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (21.3) Requirement already satisfied: pyDeprecate==0.3.1 in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (0.3.1) Requirement already satisfied: aiohttp in /usr/local/lib/python3.6/dist-packages (from fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (3.8.6) Requirement already satisfied: imageio>=2.3.0 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (2.9.0) Requirement already satisfied: tifffile>=2019.7.26 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (2020.9.3) Requirement already satisfied: PyWavelets>=1.1.1 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (1.1.1) Requirement already satisfied: networkx>=2.0 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (2.5.1) Requirement already satisfied: decorator<5,>=4.3 in /usr/local/lib/python3.6/dist-packages (from networkx>=2.0->scikit-image->-r requirements.txt (line 56)) (4.4.2) Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.2.0) Requirement already satisfied: asynctest==0.13.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (0.13.0) Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (5.2.0) Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (22.2.0) Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.2.0) Requirement already satisfied: async-timeout<5.0,>=4.0.0a3 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (4.0.2) Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.7.2) Requirement already satisfied: idna-ssl>=1.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.1.0) Installing collected packages: tensorboard Attempting uninstall: tensorboard Found existing installation: tensorboard 2.10.1 Uninstalling tensorboard-2.10.1: Successfully uninstalled tensorboard-2.10.1 Successfully installed tensorboard-2.2.1 WARNING: You are using pip version 21.0.1; however, version 21.3.1 is available. You should consider upgrading via the '/usr/bin/python -m pip install --upgrade pip' command. root@81f78440247d:/home/workspace/StyleGenForLabels# pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple, https://download.pytorch.org/whl/cu113 Requirement already satisfied: torch in /usr/local/lib/python3.6/dist-packages (1.8.1+cu111) Requirement already satisfied: torchvision in /usr/local/lib/python3.6/dist-packages (0.9.1+cu111) Requirement already satisfied: torchaudio in /usr/local/lib/python3.6/dist-packages (0.8.1) Requirement already satisfied: dataclasses in /usr/local/lib/python3.6/dist-packages (from torch) (0.8) Requirement already satisfied: typing-extensions in /usr/local/lib/python3.6/dist-packages (from torch) (4.1.1) Requirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from torch) (1.18.4) Requirement already satisfied: pillow>=4.1.1 in /usr/local/lib/python3.6/dist-packages (from torchvision) (7.1.2)
I have done installations using the requirements.txt file which is provided with the repo. (p.s. installed libraries details are added in the end)
root@81f78440247d:/home/workspace/StyleGenForLabels/pixel2style2pixel# python scripts/train.py \
root@81f78440247d:/home/workspace/StyleGenForLabels# pip install -r requirements.txt Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple Requirement already satisfied: absl-py==1.0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 1)) (1.0.0) Requirement already satisfied: cachetools==4.2.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 2)) (4.2.4) Requirement already satisfied: certifi==2020.4.5.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 3)) (2020.4.5.1) Requirement already satisfied: charset-normalizer==2.0.12 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 4)) (2.0.12) Requirement already satisfied: click==8.0.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 5)) (8.0.4) Requirement already satisfied: cycler==0.11.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 6)) (0.11.0) Requirement already satisfied: dataclasses==0.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 7)) (0.8) Requirement already satisfied: docker-pycreds==0.4.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 8)) (0.4.0) Requirement already satisfied: future==0.18.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 9)) (0.18.2) Requirement already satisfied: gitdb==4.0.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 10)) (4.0.9) Requirement already satisfied: GitPython==3.1.18 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 11)) (3.1.18) Requirement already satisfied: google-auth==1.35.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 12)) (1.35.0) Requirement already satisfied: google-auth-oauthlib==0.4.6 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 13)) (0.4.6) Requirement already satisfied: grpcio==1.44.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 14)) (1.44.0) Requirement already satisfied: idna==3.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 15)) (3.3) Requirement already satisfied: importlib-metadata==4.8.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 16)) (4.8.3) Requirement already satisfied: joblib==1.1.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 17)) (1.1.0) Requirement already satisfied: kiwisolver==1.3.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 18)) (1.3.1) Requirement already satisfied: lmdb==1.3.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 19)) (1.3.0) Requirement already satisfied: Markdown==3.3.6 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 20)) (3.3.6) Requirement already satisfied: matplotlib==3.2.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 21)) (3.2.1) Requirement already satisfied: ninja==1.11.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 22)) (1.11.1) Requirement already satisfied: numpy==1.18.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 23)) (1.18.4) Requirement already satisfied: oauthlib==3.2.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 24)) (3.2.0) Requirement already satisfied: opencv-python==4.2.0.34 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 25)) (4.2.0.34) Requirement already satisfied: pathtools==0.1.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 26)) (0.1.2) Requirement already satisfied: Pillow==7.1.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 27)) (7.1.2) Requirement already satisfied: promise==2.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 28)) (2.3) Requirement already satisfied: protobuf==3.19.4 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 29)) (3.19.4) Requirement already satisfied: psutil==5.9.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 30)) (5.9.1) Requirement already satisfied: pyasn1==0.4.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 31)) (0.4.8) Requirement already satisfied: pyasn1-modules==0.2.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 32)) (0.2.8) Requirement already satisfied: pyparsing==3.0.7 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 33)) (3.0.7) Requirement already satisfied: python-dateutil==2.8.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 34)) (2.8.2) Requirement already satisfied: PyYAML==6.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 35)) (6.0) Requirement already satisfied: requests==2.27.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 36)) (2.27.1) Requirement already satisfied: requests-oauthlib==1.3.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 37)) (1.3.1) Requirement already satisfied: rsa==4.8 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 38)) (4.8) Requirement already satisfied: scikit-learn==0.24.2 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 39)) (0.24.2) Requirement already satisfied: scipy==1.4.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 40)) (1.4.1) Requirement already satisfied: sentry-sdk==1.9.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 41)) (1.9.0) Requirement already satisfied: setproctitle==1.2.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 42)) (1.2.3) Requirement already satisfied: shortuuid==1.0.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 43)) (1.0.9) Requirement already satisfied: six==1.16.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 44)) (1.16.0) Requirement already satisfied: sklearn==0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 45)) (0.0) Requirement already satisfied: smmap==5.0.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 46)) (5.0.0) Collecting tensorboard==2.2.1 Using cached https://pypi.tuna.tsinghua.edu.cn/packages/1d/fd/4f3ca1516cbb3713259ef229abd9314bba0077ef6070285dde0dd1ed21b2/tensorboard-2.2.1-py3-none-any.whl (3.0 MB) Requirement already satisfied: tensorboard-plugin-wit==1.8.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 48)) (1.8.1) Requirement already satisfied: threadpoolctl==3.1.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 49)) (3.1.0) Requirement already satisfied: tqdm==4.46.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 50)) (4.46.0) Requirement already satisfied: typing-extensions==4.1.1 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 51)) (4.1.1) Requirement already satisfied: urllib3==1.26.9 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 52)) (1.26.9) Requirement already satisfied: wandb==0.13.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 53)) (0.13.0) Requirement already satisfied: Werkzeug==2.0.3 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 54)) (2.0.3) Requirement already satisfied: zipp==3.6.0 in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 55)) (3.6.0) Requirement already satisfied: scikit-image in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 56)) (0.17.2) Requirement already satisfied: torchmetrics in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 57)) (0.8.2) Requirement already satisfied: oyaml in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 58)) (1.0) Requirement already satisfied: pytorch_lightning in /usr/local/lib/python3.6/dist-packages (from -r requirements.txt (line 59)) (1.5.10) Requirement already satisfied: setuptools>=40.3.0 in /usr/local/lib/python3.6/dist-packages (from google-auth==1.35.0->-r requirements.txt (line 12)) (59.5.0) Requirement already satisfied: wheel>=0.26 in /usr/lib/python3/dist-packages (from tensorboard==2.2.1->-r requirements.txt (line 47)) (0.30.0) Requirement already satisfied: fsspec[http]!=2021.06.0,>=2021.05.0 in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (2022.1.0) Requirement already satisfied: torch>=1.7.* in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (1.8.1+cu111) Requirement already satisfied: packaging>=17.0 in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (21.3) Requirement already satisfied: pyDeprecate==0.3.1 in /usr/local/lib/python3.6/dist-packages (from pytorch_lightning->-r requirements.txt (line 59)) (0.3.1) Requirement already satisfied: aiohttp in /usr/local/lib/python3.6/dist-packages (from fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (3.8.6) Requirement already satisfied: imageio>=2.3.0 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (2.9.0) Requirement already satisfied: tifffile>=2019.7.26 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (2020.9.3) Requirement already satisfied: PyWavelets>=1.1.1 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (1.1.1) Requirement already satisfied: networkx>=2.0 in /usr/local/lib/python3.6/dist-packages (from scikit-image->-r requirements.txt (line 56)) (2.5.1) Requirement already satisfied: decorator<5,>=4.3 in /usr/local/lib/python3.6/dist-packages (from networkx>=2.0->scikit-image->-r requirements.txt (line 56)) (4.4.2) Requirement already satisfied: aiosignal>=1.1.2 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.2.0) Requirement already satisfied: asynctest==0.13.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (0.13.0) Requirement already satisfied: multidict<7.0,>=4.5 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (5.2.0) Requirement already satisfied: attrs>=17.3.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (22.2.0) Requirement already satisfied: frozenlist>=1.1.1 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.2.0) Requirement already satisfied: async-timeout<5.0,>=4.0.0a3 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (4.0.2) Requirement already satisfied: yarl<2.0,>=1.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.7.2) Requirement already satisfied: idna-ssl>=1.0 in /usr/local/lib/python3.6/dist-packages (from aiohttp->fsspec[http]!=2021.06.0,>=2021.05.0->pytorch_lightning->-r requirements.txt (line 59)) (1.1.0) Installing collected packages: tensorboard Attempting uninstall: tensorboard Found existing installation: tensorboard 2.10.1 Uninstalling tensorboard-2.10.1: Successfully uninstalled tensorboard-2.10.1 Successfully installed tensorboard-2.2.1 WARNING: You are using pip version 21.0.1; however, version 21.3.1 is available. You should consider upgrading via the '/usr/bin/python -m pip install --upgrade pip' command. root@81f78440247d:/home/workspace/StyleGenForLabels# pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 Looking in indexes: https://pypi.tuna.tsinghua.edu.cn/simple, https://download.pytorch.org/whl/cu113 Requirement already satisfied: torch in /usr/local/lib/python3.6/dist-packages (1.8.1+cu111) Requirement already satisfied: torchvision in /usr/local/lib/python3.6/dist-packages (0.9.1+cu111) Requirement already satisfied: torchaudio in /usr/local/lib/python3.6/dist-packages (0.8.1) Requirement already satisfied: dataclasses in /usr/local/lib/python3.6/dist-packages (from torch) (0.8) Requirement already satisfied: typing-extensions in /usr/local/lib/python3.6/dist-packages (from torch) (4.1.1) Requirement already satisfied: numpy in /usr/local/lib/python3.6/dist-packages (from torch) (1.18.4) Requirement already satisfied: pillow>=4.1.1 in /usr/local/lib/python3.6/dist-packages (from torchvision) (7.1.2)