Bug fix release with updated binaries for Python 3.9 and cuDNN 8.0.5. Source:https://github.com/pytorch/pytorch/releases/tag/v1.7.1
PyTorch 1.7.1 Release NotesNew Features
Critical Fixes
Other Fixes
New FeaturesAdd 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.
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.
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)
BuildFix 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)
Misctorch.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 FixesTensor 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)