saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/caffe2
History
pbialecki 3c7b2b730f use libcusolver_lapack_static.a for CUDA>=12 (#98072) 
Needed for https://github.com/pytorch/builder/pull/1374 to enable nightly CUDA12.1 builds.

From the cuSOLVER release notes (https://docs.nvidia.com/cuda/cusolver/index.html#link-third-party-lapack-library):
> The `liblapack_static.a` library is deprecated and will be removed in the next major release. Use the `libcusolver_lapack_static.a` instead.

Note that "next major release" corresponds to CUDA 12, not 13.
The fix was verified locally on an H100 using https://github.com/pytorch/builder/pull/1374 and pip wheels were properly built:
```
>>> torch.version.cuda
'12.1'
>>> torch.backends.cudnn.version()
8801
>>> conv =nn.Conv2d(3, 3, 3).cuda()
>>> x = torch.randn(1, 3, 224, 224).cuda()
>>> out = conv(x)
>>> out.sum()
tensor(5386.9219, device='cuda:0', grad_fn=<SumBackward0>)
```

CC @malfet @atalman @ngimel
Pull Request resolved: https://github.com/pytorch/pytorch/pull/98072
Approved by: https://github.com/malfet, https://github.com/atalman
2023-03-31 20:47:53 +00:00
..
contrib Follow up on CUDA 12 support for PyTorch/Caffe2 (#95582) 2023-02-27 04:39:56 +00:00
core remove unused CAFFE2_VERSION macros (#97337) 2023-03-24 16:02:35 +00:00
cuda_rtc
db
distributed
experiments
ideep
image
mobile
mpi
observers
onnx
operators
opt some reference and move fixes (#95942) 2023-03-10 03:44:09 +00:00
perfkernels [BE] Enforce sign-compare (#96723) 2023-03-15 06:04:20 +00:00
predictor
proto extract torch.proto to its own library (#97614) 2023-03-30 10:35:03 +00:00
python Add DLPack support for XPU backend by mapping to kDLOneAPI in DLPack … (#94968) 2023-03-30 04:32:15 +00:00
quantization
queue
serialize [BE] Enforce sign-compare (#96723) 2023-03-15 06:04:20 +00:00
sgd
share
test
transforms
utils remove dead proto_convert library (#97598) 2023-03-27 21:19:29 +00:00
video
.clang-format
__init__.py
BUILD_MODE.bzl
CMakeLists.txt use libcusolver_lapack_static.a for CUDA>=12 (#98072) 2023-03-31 20:47:53 +00:00
README.md
release-notes.md
requirements.txt
unexported_symbols.lds
VERSION_NUMBER
version_script.lds

README.md

Caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework. Building on the original Caffe, Caffe2 is designed with expression, speed, and modularity in mind.

Questions and Feedback

Please use GitHub issues (https://github.com/pytorch/pytorch/issues) to ask questions, report bugs, and request new features.

Further Resources on Caffe2.ai