{"id":15681274,"url":"https://github.com/pinto0309/pytorch-build","last_synced_at":"2025-05-07T12:11:58.953Z","repository":{"id":50994179,"uuid":"339936263","full_name":"PINTO0309/PyTorch-build","owner":"PINTO0309","description":"Provide Docker build sequences of PyTorch for various environments.","archived":false,"fork":false,"pushed_at":"2021-05-26T08:48:06.000Z","size":25,"stargazers_count":16,"open_issues_count":0,"forks_count":4,"subscribers_count":2,"default_branch":"main","last_synced_at":"2025-05-07T12:11:51.274Z","etag":null,"topics":["cuda","cudnn","docker","pytorch"],"latest_commit_sha":null,"homepage":"https://qiita.com/PINTO","language":"Dockerfile","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"mit","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/PINTO0309.png","metadata":{"files":{"readme":"README.md","changelog":null,"contributing":null,"funding":null,"license":"LICENSE","code_of_conduct":null,"threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":null,"support":null}},"created_at":"2021-02-18T04:34:28.000Z","updated_at":"2025-04-01T01:46:14.000Z","dependencies_parsed_at":"2022-09-18T21:38:33.980Z","dependency_job_id":null,"html_url":"https://github.com/PINTO0309/PyTorch-build","commit_stats":null,"previous_names":[],"tags_count":3,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/PINTO0309%2FPyTorch-build","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/PINTO0309%2FPyTorch-build/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/PINTO0309%2FPyTorch-build/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/PINTO0309%2FPyTorch-build/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/PINTO0309","download_url":"https://codeload.github.com/PINTO0309/PyTorch-build/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":252873898,"owners_count":21817715,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["cuda","cudnn","docker","pytorch"],"created_at":"2024-10-03T16:51:52.053Z","updated_at":"2025-05-07T12:11:58.918Z","avatar_url":"https://github.com/PINTO0309.png","language":"Dockerfile","funding_links":[],"categories":[],"sub_categories":[],"readme":"# PyTorch-build\nProvide Docker build sequences of PyTorch for various environments.\n**https://github.com/pytorch/pytorch**\n\n## 1. Docker Image Environment\n- Ubuntu 20.04 x86_64\n- CUDA 11.3\n- cuDNN 8.2\n- PyTorch v1.8.1 (Build from source code. It will be downloaded automatically during docker build.)\n- TorchVision v0.9.1\n- TorchAudio v0.8.1\n\n## 2. Default build parameters\n\n```\n--\n-- ******** Summary ********\n-- General:\n--   CMake version         : 3.18.4\n--   CMake command         : /usr/local/lib/python3.8/dist-packages/cmake/data/bin/cmake\n--   System                : Linux\n--   C++ compiler          : /usr/bin/c++\n--   C++ compiler id       : GNU\n--   C++ compiler version  : 9.3.0\n--   CXX flags             :  -D_GLIBCXX_USE_CXX11_ABI=1 -Wno-deprecated -fvisibility-inlines-hidden -DUSE_PTHREADPOOL -fopenmp -DNDEBUG -DUSE_KINETO -DUSE_FBGEMM -DUSE_QNNPACK -DUSE_PYTORCH_QNNPACK -DUSE_XNNPACK -O2 -fPIC -Wno-narrowing -Wall -Wextra -Werror=return-type -Wno-missing-field-initializers -Wno-type-limits -Wno-array-bounds -Wno-unknown-pragmas -Wno-sign-compare -Wno-unused-parameter -Wno-unused-variable -Wno-unused-function -Wno-unused-result -Wno-unused-local-typedefs -Wno-strict-overflow -Wno-strict-aliasing -Wno-error=deprecated-declarations -Wno-stringop-overflow -Wno-psabi -Wno-error=pedantic -Wno-error=redundant-decls -Wno-error=old-style-cast -fdiagnostics-color=always -faligned-new -Wno-unused-but-set-variable -Wno-maybe-uninitialized -fno-math-errno -fno-trapping-math -Werror=format -Werror=cast-function-type -Wno-stringop-overflow\n--   Build type            : Release\n--   Compile definitions   : ONNX_ML=1;ONNXIFI_ENABLE_EXT=1;ONNX_NAMESPACE=onnx_torch;HAVE_MMAP=1;_FILE_OFFSET_BITS=64;HAVE_SHM_OPEN=1;HAVE_SHM_UNLINK=1;HAVE_MALLOC_USABLE_SIZE=1;USE_EXTERNAL_MZCRC;MINIZ_DISABLE_ZIP_READER_CRC32_CHECKS\n--   CMAKE_PREFIX_PATH     : /usr/lib/python3/dist-packages;/usr/local/cuda;/usr/local/cuda\n--   CMAKE_INSTALL_PREFIX  : /pytorch/torch\n--\n--   TORCH_VERSION         : 1.8.0\n--   CAFFE2_VERSION        : 1.8.0\n--   BUILD_CAFFE2          : ON\n--   BUILD_CAFFE2_OPS      : ON\n--   BUILD_CAFFE2_MOBILE   : OFF\n--   BUILD_STATIC_RUNTIME_BENCHMARK: OFF\n--   BUILD_TENSOREXPR_BENCHMARK: OFF\n--   BUILD_BINARY          : OFF\n--   BUILD_CUSTOM_PROTOBUF : ON\n--     Link local protobuf : ON\n--   BUILD_DOCS            : OFF\n--   BUILD_PYTHON          : True\n--     Python version      : 3.8.5\n--     Python executable   : /usr/bin/python3\n--     Pythonlibs version  : 3.8.5\n--     Python library      : /usr/lib/libpython3.8.so.1.0\n--     Python includes     : /usr/include/python3.8\n--     Python site-packages: lib/python3/dist-packages\n--   BUILD_SHARED_LIBS     : ON\n--   CAFFE2_USE_MSVC_STATIC_RUNTIME     : OFF\n--   BUILD_TEST            : True\n--   BUILD_JNI             : OFF\n--   BUILD_MOBILE_AUTOGRAD : OFF\n--   INTERN_BUILD_MOBILE   :\n--   USE_BLAS              : 1\n--     BLAS                : generic\n--   USE_LAPACK            : 1\n--     LAPACK              : generic\n--   USE_ASAN              : OFF\n--   USE_CPP_CODE_COVERAGE : OFF\n--   USE_CUDA              : ON\n--     Split CUDA          : OFF\n--     CUDA static link    : OFF\n--     USE_CUDNN           : ON\n--     CUDA version        : 11.3\n--     cuDNN version       : 8.2.0\n--     CUDA root directory : /usr/local/cuda\n--     CUDA library        : /usr/local/cuda/lib64/stubs/libcuda.so\n--     cudart library      : /usr/local/cuda/lib64/libcudart.so\n--     cublas library      : /usr/local/cuda/lib64/libcublas.so\n--     cufft library       : /usr/local/cuda/lib64/libcufft.so\n--     curand library      : /usr/local/cuda/lib64/libcurand.so\n--     cuDNN library       : /usr/lib/x86_64-linux-gnu/libcudnn.so\n--     nvrtc               : /usr/local/cuda/lib64/libnvrtc.so\n--     CUDA include path   : /usr/local/cuda/include\n--     NVCC executable     : /usr/local/cuda/bin/nvcc\n--     NVCC flags          : -Xfatbin;-compress-all;-DONNX_NAMESPACE=onnx_torch;-gencode;arch=compute_35,code=sm_35;-gencode;arch=compute_50,code=sm_50;-gencode;arch=compute_52,code=sm_52;-gencode;arch=compute_60,code=sm_60;-gencode;arch=compute_61,code=sm_61;-gencode;arch=compute_70,code=sm_70;-gencode;arch=compute_75,code=sm_75;-gencode;arch=compute_80,code=sm_80;-gencode;arch=compute_86,code=sm_86;-gencode;arch=compute_86,code=compute_86;-Xcudafe;--diag_suppress=cc_clobber_ignored,--diag_suppress=integer_sign_change,--diag_suppress=useless_using_declaration,--diag_suppress=set_but_not_used,--diag_suppress=field_without_dll_interface,--diag_suppress=base_class_has_different_dll_interface,--diag_suppress=dll_interface_conflict_none_assumed,--diag_suppress=dll_interface_conflict_dllexport_assumed,--diag_suppress=implicit_return_from_non_void_function,--diag_suppress=unsigned_compare_with_zero,--diag_suppress=declared_but_not_referenced,--diag_suppress=bad_friend_decl;-std=c++14;-Xcompiler;-fPIC;--expt-relaxed-constexpr;--expt-extended-lambda;-Wno-deprecated-gpu-targets;--expt-extended-lambda;-Xcompiler;-fPIC;-DCUDA_HAS_FP16=1;-D__CUDA_NO_HALF_OPERATORS__;-D__CUDA_NO_HALF_CONVERSIONS__;-D__CUDA_NO_BFLOAT16_CONVERSIONS__;-D__CUDA_NO_HALF2_OPERATORS__\n--     CUDA host compiler  : /usr/bin/cc\n--     NVCC --device-c     : OFF\n--     USE_TENSORRT        : OFF\n--   USE_ROCM              : OFF\n--   USE_EIGEN_FOR_BLAS    : ON\n--   USE_FBGEMM            : ON\n--     USE_FAKELOWP          : OFF\n--   USE_KINETO            : ON\n--   USE_FFMPEG            : OFF\n--   USE_GFLAGS            : OFF\n--   USE_GLOG              : OFF\n--   USE_LEVELDB           : OFF\n--   USE_LITE_PROTO        : OFF\n--   USE_LMDB              : OFF\n--   USE_METAL             : OFF\n--   USE_PYTORCH_METAL     : OFF\n--   USE_FFTW              : OFF\n--   USE_MKL               : OFF\n--   USE_MKLDNN            : ON\n--   USE_MKLDNN_CBLAS      : OFF\n--   USE_NCCL              : OFF\n--     USE_SYSTEM_NCCL     : OFF\n--   USE_NNPACK            : ON\n--   USE_NUMPY             : ON\n--   USE_OBSERVERS         : ON\n--   USE_OPENCL            : OFF\n--   USE_OPENCV            : OFF\n--   USE_OPENMP            : ON\n--   USE_TBB               : OFF\n--   USE_VULKAN            : OFF\n--   USE_PROF              : OFF\n--   USE_QNNPACK           : ON\n--   USE_PYTORCH_QNNPACK   : ON\n--   USE_REDIS             : OFF\n--   USE_ROCKSDB           : OFF\n--   USE_ZMQ               : OFF\n--   USE_DISTRIBUTED       : ON\n--     USE_MPI             : OFF\n--     USE_GLOO            : ON\n--     USE_TENSORPIPE      : ON\n--   USE_DEPLOY           : OFF\n--   Public Dependencies  : Threads::Threads;caffe2::mkldnn\n--   Private Dependencies : pthreadpool;cpuinfo;qnnpack;pytorch_qnnpack;nnpack;XNNPACK;fbgemm;fp16;gloo;tensorpipe;aten_op_header_gen;foxi_loader;rt;fmt::fmt-header-only;kineto;gcc_s;gcc;dl\n-- Configuring done\n-- Generating done\n-- Build files have been written to: /pytorch/build\n```\n\n## 3. Usage - Docker Build\nYou can customize the Dockerfile to build and run your own container images on your own.\n```bash\n$ version=1.8.1\n$ git clone -b ${version} https://github.com/PINTO0309/PyTorch-build.git\n$ cd PyTorch-build\n$ docker build -t pinto0309/pytorch-build:11.3.0-cudnn8-devel-ubuntu20.04 .\n\n$ docker run --gpus all -it --rm \\\n    -v `pwd`:/workspace \\\n    -e LOCAL_UID=$(id -u $USER) \\\n    -e LOCAL_GID=$(id -g $USER) \\\n    pinto0309/pytorch-build:11.3.0-cudnn8-devel-ubuntu20.04 bash\n```\n\n## 4. Usage - Docker Pull / Run\nYou can download and run a pre-built container image from Docker Hub.\n```bash\n$ docker run --gpus all -it --rm \\\n    -v `pwd`:/workspace \\\n    -e LOCAL_UID=$(id -u $USER) \\\n    -e LOCAL_GID=$(id -g $USER) \\\n    pinto0309/pytorch-build:11.3.0-cudnn8-devel-ubuntu20.04 bash\n```","project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fpinto0309%2Fpytorch-build","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fpinto0309%2Fpytorch-build","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fpinto0309%2Fpytorch-build/lists"}