Skip to content
Choose a tag to compare

Bug fix release with updated binaries for Python 3.9 and cuDNN 8.0.5

@albanD albanD released this
· 1 commit to release/1.7 since this release
Choose a tag to compare

PyTorch 1.7.1 Release Notes

  • New Features
  • Critical Fixes
  • Other Fixes

New Features

Add Python 3.9 binaries for linux and macOS (#48133) and Windows (#48218)

NOTE: Conda installs for Python 3.9 will require the conda-forge channel, example:
conda install -y -c pytorch -c conda-forge pytorch.

Upgrade CUDA binaries to use cuDNN 8.0.5 (builder repo #571)

This upgrade fix regressions on Ampere cards introduced in cuDNN 8.0.4.
It will improve performance for 3090 RTX cards, and may improve performance in other RTX-30 series card.

Critical Fixes

Python 3.9

  • Use custom version of pybind11 to work around Python 3.9 issues (#48312)
  • Fix jit Python 3.9 parsing (#48744)
  • Fix cpp_extension to work with Python 3.9 (#48768)


  • Fix cpp_extension to properly handle env variable on Windows (#48937)
  • Properly package libomp.dylib for macOS binaries (#48337)
  • Fix build for statically linked OpenBLAS on aarch64 (#48819)


  • torch.sqrt: fix wrong output values for very large complex input (#48216)
  • max_pool1d: fix for discontiguous inputs (#48219)
  • collect_env: fix detection of DEBUG flag (#48319)
  • collect_env: Fix to work when PyTorch is not installed (#48311)
  • Fix amp memory usage when running in no_grad() mode (#48936)
  • nn.ParameterList and nn.ParameterDict: Remove spurious warnings (#48215)
  • Tensor Expression fuser bugfixes (#48137)

Other Fixes

  • Tensor Expression fix for CUDA 11.0 (#48309)
  • torch.overrides: doc fix (#47843)
  • torch.max: Fix output type for Tensor subclasses (#47735)
  • torch.mul: Add support for boolean Tensors (#48310)
  • Add user friendly error when trying to compile from source with Python 2 (#48317)