Skip to content

Instantly share code, notes, and snippets.

@msaroufim
Created March 29, 2023 23:42
Show Gist options
  • Save msaroufim/045f4686fdb9fc46571b451e160563a3 to your computer and use it in GitHub Desktop.
Save msaroufim/045f4686fdb9fc46571b451e160563a3 to your computer and use it in GitHub Desktop.
(download) ubuntu@ip-172-31-38-220:~/serve/docker$ python check_sizes.py
Collecting torch==1.11.0
Using cached torch-1.11.0-cp310-cp310-manylinux1_x86_64.whl (750.6 MB)
Collecting typing-extensions
Using cached typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Saved /tmp/tmp0ra0n5j9/torch-1.11.0-cp310-cp310-manylinux1_x86_64.whl
Saved /tmp/tmp0ra0n5j9/typing_extensions-4.5.0-py3-none-any.whl
Successfully downloaded torch typing-extensions
Collecting torch==1.12.0
Using cached torch-1.12.0-cp310-cp310-manylinux1_x86_64.whl (776.3 MB)
Collecting typing-extensions
Using cached typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Saved /tmp/tmpfe_lwomg/torch-1.12.0-cp310-cp310-manylinux1_x86_64.whl
Saved /tmp/tmpfe_lwomg/typing_extensions-4.5.0-py3-none-any.whl
Successfully downloaded torch typing-extensions
Collecting torch==1.12.1
Using cached torch-1.12.1-cp310-cp310-manylinux1_x86_64.whl (776.3 MB)
Collecting typing-extensions
Using cached typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Saved /tmp/tmptn887m_r/torch-1.12.1-cp310-cp310-manylinux1_x86_64.whl
Saved /tmp/tmptn887m_r/typing_extensions-4.5.0-py3-none-any.whl
Successfully downloaded torch typing-extensions
Collecting torch==1.13.0
Using cached torch-1.13.0-cp310-cp310-manylinux1_x86_64.whl (890.1 MB)
Collecting typing-extensions
Using cached typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Collecting nvidia-cudnn-cu11==8.5.0.96
Using cached nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl (557.1 MB)
Collecting nvidia-cuda-nvrtc-cu11==11.7.99
Using cached nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl (21.0 MB)
Collecting nvidia-cublas-cu11==11.10.3.66
Using cached nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl (317.1 MB)
Collecting nvidia-cuda-runtime-cu11==11.7.99
Using cached nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl (849 kB)
Collecting wheel
Using cached wheel-0.40.0-py3-none-any.whl (64 kB)
Collecting setuptools
Using cached setuptools-67.6.1-py3-none-any.whl (1.1 MB)
Saved /tmp/tmpca7kdlfx/torch-1.13.0-cp310-cp310-manylinux1_x86_64.whl
Saved /tmp/tmpca7kdlfx/nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpca7kdlfx/nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpca7kdlfx/nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpca7kdlfx/nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpca7kdlfx/typing_extensions-4.5.0-py3-none-any.whl
Saved /tmp/tmpca7kdlfx/wheel-0.40.0-py3-none-any.whl
Saved /tmp/tmpca7kdlfx/setuptools-67.6.1-py3-none-any.whl
Successfully downloaded torch nvidia-cublas-cu11 nvidia-cuda-nvrtc-cu11 nvidia-cuda-runtime-cu11 nvidia-cudnn-cu11 typing-extensions wheel setuptools
Collecting torch==1.13.1
Using cached torch-1.13.1-cp310-cp310-manylinux1_x86_64.whl (887.5 MB)
Collecting nvidia-cuda-nvrtc-cu11==11.7.99
Using cached nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl (21.0 MB)
Collecting typing-extensions
Using cached typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Collecting nvidia-cuda-runtime-cu11==11.7.99
Using cached nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl (849 kB)
Collecting nvidia-cudnn-cu11==8.5.0.96
Using cached nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl (557.1 MB)
Collecting nvidia-cublas-cu11==11.10.3.66
Using cached nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl (317.1 MB)
Collecting setuptools
Using cached setuptools-67.6.1-py3-none-any.whl (1.1 MB)
Collecting wheel
Using cached wheel-0.40.0-py3-none-any.whl (64 kB)
Saved /tmp/tmpmab82da2/torch-1.13.1-cp310-cp310-manylinux1_x86_64.whl
Saved /tmp/tmpmab82da2/nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpmab82da2/nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpmab82da2/nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpmab82da2/nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpmab82da2/typing_extensions-4.5.0-py3-none-any.whl
Saved /tmp/tmpmab82da2/wheel-0.40.0-py3-none-any.whl
Saved /tmp/tmpmab82da2/setuptools-67.6.1-py3-none-any.whl
Successfully downloaded torch nvidia-cublas-cu11 nvidia-cuda-nvrtc-cu11 nvidia-cuda-runtime-cu11 nvidia-cudnn-cu11 typing-extensions wheel setuptools
Collecting torch==2.0.0
Using cached torch-2.0.0-cp310-cp310-manylinux1_x86_64.whl (619.9 MB)
Collecting jinja2
Using cached Jinja2-3.1.2-py3-none-any.whl (133 kB)
Collecting nvidia-cusolver-cu11==11.4.0.1
Using cached nvidia_cusolver_cu11-11.4.0.1-2-py3-none-manylinux1_x86_64.whl (102.6 MB)
Collecting nvidia-cuda-runtime-cu11==11.7.99
Using cached nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl (849 kB)
Collecting triton==2.0.0
Using cached triton-2.0.0-1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (63.3 MB)
Collecting sympy
Using cached sympy-1.11.1-py3-none-any.whl (6.5 MB)
Collecting nvidia-cuda-cupti-cu11==11.7.101
Using cached nvidia_cuda_cupti_cu11-11.7.101-py3-none-manylinux1_x86_64.whl (11.8 MB)
Collecting nvidia-cufft-cu11==10.9.0.58
Using cached nvidia_cufft_cu11-10.9.0.58-py3-none-manylinux1_x86_64.whl (168.4 MB)
Collecting nvidia-curand-cu11==10.2.10.91
Using cached nvidia_curand_cu11-10.2.10.91-py3-none-manylinux1_x86_64.whl (54.6 MB)
Collecting filelock
Using cached filelock-3.10.7-py3-none-any.whl (10 kB)
Collecting networkx
Using cached networkx-3.0-py3-none-any.whl (2.0 MB)
Collecting nvidia-cuda-nvrtc-cu11==11.7.99
Using cached nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl (21.0 MB)
Collecting nvidia-nccl-cu11==2.14.3
Using cached nvidia_nccl_cu11-2.14.3-py3-none-manylinux1_x86_64.whl (177.1 MB)
Collecting nvidia-nvtx-cu11==11.7.91
Using cached nvidia_nvtx_cu11-11.7.91-py3-none-manylinux1_x86_64.whl (98 kB)
Collecting nvidia-cudnn-cu11==8.5.0.96
Using cached nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl (557.1 MB)
Collecting typing-extensions
Using cached typing_extensions-4.5.0-py3-none-any.whl (27 kB)
Collecting nvidia-cusparse-cu11==11.7.4.91
Using cached nvidia_cusparse_cu11-11.7.4.91-py3-none-manylinux1_x86_64.whl (173.2 MB)
Collecting nvidia-cublas-cu11==11.10.3.66
Using cached nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl (317.1 MB)
Collecting setuptools
Using cached setuptools-67.6.1-py3-none-any.whl (1.1 MB)
Collecting wheel
Using cached wheel-0.40.0-py3-none-any.whl (64 kB)
Collecting lit
Using cached lit-16.0.0.tar.gz (144 kB)
Preparing metadata (setup.py) ... done
Collecting cmake
Using cached cmake-3.26.1-py2.py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl (24.0 MB)
Collecting MarkupSafe>=2.0
Using cached MarkupSafe-2.1.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (25 kB)
Collecting mpmath>=0.19
Using cached mpmath-1.3.0-py3-none-any.whl (536 kB)
Saved /tmp/tmpufs2krk3/torch-2.0.0-cp310-cp310-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cublas_cu11-11.10.3.66-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cuda_cupti_cu11-11.7.101-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cuda_nvrtc_cu11-11.7.99-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cuda_runtime_cu11-11.7.99-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cudnn_cu11-8.5.0.96-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cufft_cu11-10.9.0.58-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_curand_cu11-10.2.10.91-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cusolver_cu11-11.4.0.1-2-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_cusparse_cu11-11.7.4.91-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_nccl_cu11-2.14.3-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/nvidia_nvtx_cu11-11.7.91-py3-none-manylinux1_x86_64.whl
Saved /tmp/tmpufs2krk3/triton-2.0.0-1-cp310-cp310-manylinux2014_x86_64.manylinux_2_17_x86_64.whl
Saved /tmp/tmpufs2krk3/filelock-3.10.7-py3-none-any.whl
Saved /tmp/tmpufs2krk3/Jinja2-3.1.2-py3-none-any.whl
Saved /tmp/tmpufs2krk3/networkx-3.0-py3-none-any.whl
Saved /tmp/tmpufs2krk3/sympy-1.11.1-py3-none-any.whl
Saved /tmp/tmpufs2krk3/typing_extensions-4.5.0-py3-none-any.whl
Saved /tmp/tmpufs2krk3/MarkupSafe-2.1.2-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl
Saved /tmp/tmpufs2krk3/mpmath-1.3.0-py3-none-any.whl
Saved /tmp/tmpufs2krk3/cmake-3.26.1-py2.py3-none-manylinux2014_x86_64.manylinux_2_17_x86_64.whl
Saved /tmp/tmpufs2krk3/lit-16.0.0.tar.gz
Saved /tmp/tmpufs2krk3/wheel-0.40.0-py3-none-any.whl
Saved /tmp/tmpufs2krk3/setuptools-67.6.1-py3-none-any.whl
Successfully downloaded torch nvidia-cublas-cu11 nvidia-cuda-cupti-cu11 nvidia-cuda-nvrtc-cu11 nvidia-cuda-runtime-cu11 nvidia-cudnn-cu11 nvidia-cufft-cu11 nvidia-curand-cu11 nvidia-cusolver-cu11 nvidia-cusparse-cu11 nvidia-nccl-cu11 nvidia-nvtx-cu11 triton filelock jinja2 networkx s
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment