解决CUDA error: no kernel image is available for execution on the device
1. 电脑配置GPU 3080 算力8.6CUDA 11.1CUDNN 8.2.0conda 4.9.2python 3.8.52. 问题描述首先在pytroch官网根据电脑相应配置使用pip命令pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html安装torch、torchau
原创
2021-05-24 23:55:13 ·
87562 阅读 ·
27 评论