You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
It should only contain runtime dependencies that PyTorch+domain
libraries depend on, namely:
- cudart
- cublas
- cusparse
- cufft
- curand
- nvtx
- nvrtc
- nvjpeg (for TorchVision)
This removes dependencies on NVCC, build/debug tools, etc which are not
needed for running the pytorch
Test Plan:
`conda create -n tmp -c nvidia -c malfet cuda-toolkit==11.7` and
observe that only relevant packages are installed
Fixespytorch/pytorch#91334
0 commit comments