Hi, everyone, I got this error when the program running on GPU, but everything is ok when running on the CPU

微信图片_202107282004211649×245 17.1 KB

my model is a simple seq2seq model.
Hope someone can help me, thanks a lot
here are my system infomations:
PyTorch Version (e.g., 1.0): 1.7.1
OS (e.g., Linux): Ubuntu 18.04.5 LTS (x86_64)
How you installed PyTorch (conda, pip, source): pip
Build command you used (if compiling from source):
Python version: 3.8 (64-bit runtime)
CUDA/cuDNN version: NVIDIA-SMI 440.33.01 Driver Version: 440.33.01 CUDA Version: 10.2
GPU models and configuration: RTX 2080Ti
Any other relevant information:
GCC version: (Ubuntu 7.5.0-3ubuntu1~18.04) 7.5.0
CMake version: version 3.13.2