본문 바로가기

IT/Cuda

[CUDA] NVIDIA PyTorch Container Versions

 

pytorch 1.8버전이라면 최소 11.1 이상 버전을 깔아야 한다...

 

 

 

PyTorch Release Notes :: NVIDIA Deep Learning Frameworks Documentation

The NVIDIA container image for PyTorch, release 20.12, is available on NGC. Driver Requirements Release 20.12 is based on NVIDIA CUDA 11.1.1, which requires NVIDIA Driver release 455 or later. However, if you are running on Tesla (for example, T4 or any ot

docs.nvidia.com