-
### Description
@rahulbatra85
I managed to get jax to train resnet50 using the scenic library and the latest docker images, thanks to #18747.
When I change my code from `float32` to `float16`, I …
-
ROCm?
-
I'm looking for an option to release GPU memory after Whisper tasks. Sometimes my PC shuts down due to overflowing GPU memory.
Running on Debian(Bookworm) + Docker(rocm/pytorch:rocm6.0.2_ubuntu22.04…
-
Right now some features like `test` are available only on M-series, will be great if they will be available for other platforms (Linux cuda/rocm).
-
### System Info
Docker image: rocm6.1_ubuntu22.04_py3.10_pytorch_2.4
Rocm6.1.0
Pytorch2.4
GPU: MI250
### Reproduction
Install method:
git clone --recurse https://github.com/ROCm/bitsandbytes
c…
-
Resolve needs the rocm opencl bindings, these can be installed by running the following
`sudo dnf install rocm-opencl`
-
Presently it is very hard to get a docker container to build with the rocm backend, some elements seem to fail independently during the build process.
There are other related projects with functiona…
-
### Issue type
Bug
### Have you reproduced the bug with TensorFlow Nightly?
Yes
### Source
binary
### TensorFlow version
v2.14.0-4248-g3448956e87e 2.14.0.600
### Custom code
Yes
### OS platf…
-
Am I safe to assume that DeepSpeed does not yet support ROCm 6.0? A whole lot of errors during JIT build of transformer_inference.
```
$ pip show torch
Name: torch
Version: 2.3.0+rocm6.0
```
…
-
# Feature Request Summary:
Propose adding a prompt in the initial boot greeter for users to express their preference for installing AMD's ROCm Drivers. Similar to the existing Nvidia prompt, this f…