Open baoblei opened 1 month ago
Hi @baoblei , can you check your version compatibility with the following?
# check pytorch's cuda version
python -c "import torch;print(torch.version.cuda)"
# check your cuda driver version on the top right corner
nvidia-smi
# check your cuda runtime version
nvcc -V
Thanks, I checked my version: pytorch‘s cuda version: 11.8 driver version: 12.4 runtime version: 11.8 Sorry I didn't make it clear before. This error occurred in v1.2, and I tried again in v1.1, but the same error occurred.
I'd suggest you install again: conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
This issue is stale because it has been open for 7 days with no activity.
cuda12.4