pytorch
pytorch copied to clipboard
Tensors and Dynamic neural networks in Python with strong GPU acceleration
### 🐛 Describe the bug Using torch.cuda.device() to select the GPU, and it has no effect. GPU0 is always used, expecting GPU selected to be used. On the other hand,...
### 🚀 The feature, motivation and pitch Literally.. ### Alternatives _No response_ ### Additional context _No response_
### 🐛 Describe the bug # Sample code to reproduce the problem See https://github.com/ROCm/rocBLAS/issues/1386 . This is mainly encountered after installing updating from ROCm 5.7.1 to ROCm 6.0, and compiling...
Fixes #ISSUE_NUMBER
### 🐛 Describe the bug Moving from `2.2.0.dev20231017+rocm5.6` to `2.2.0.dev20231016+rocm5.7` introduced weird behavior. As this happens with just uninstalling previous torch versions and installing new one it seems to point...
### 🐛 Describe the bug When running https://github.com/nod-ai/SHARK using ROCm 5.7 and 7900XTX, I keep getting the following error message in dmesg. ``` [ 90.648641] amdgpu 0000:2d:00.0: amdgpu: [gfxhub] page...
Fixes SWDEV-407984 and SWDEV-392430 According to Math Library team, it is expected behavior to return error when batch_count == 0. Hence I'm making the temporary workaround permanent. `PYTORCH_TEST_WITH_ROCM=1 python test/run_test.py...
### 🚀 The feature, motivation and pitch Hi, I wanna a faster transformer implemention in pytorch, and I found one in pytorch code ,whose path is pytorch/aten/src/Aten/native/transformers/cuda/, and it needs...
## 🐛 Bug The baseline code in [this colab](https://colab.research.google.com/github/davidcpage/cifar10-fast/blob/master/bag_of_tricks.ipynb#scrollTo=7WeTS0FF4iQE) provides a Pytorch example running on a V100 GPU with CUDNN to train a CIFAR10 classifier in 75 seconds. ``` epoch...
### 🐛 Describe the bug ```python $ time python -c "import torch.cuda; print(torch.cuda.is_available())" True real 3m36.487s user 0m4.613s sys 0m34.452s ``` ### Versions ``` Collecting environment information... PyTorch version: 2.0.0+rocm5.4.2...