pytorch icon indicating copy to clipboard operation
pytorch copied to clipboard

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Results 133 pytorch issues
Sort by recently updated
recently updated
newest added

### 🐛 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...

bug

### 🐛 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...