两条指令
nvidia-smi
nvcc -V
注意两条指令获得的 CUDA 版本可能并不一致,这里以 nvcc -V
显示的版本为准,具体原因暂不在本文解释。
按照表格中对应的版本进行了安装,但是仍然出现了以下报错:
RuntimeError: CUDA error: no kernel image is available for execution on the device
按照网络上常见的解释就是 torch 和 torchvision 版本不匹配,然而明明已经按照表格对应的版本进行了安装。
.whl
文件到本地。pip install
命令安装原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。
原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。
如有侵权,请联系 cloudcommunity@tencent.com 删除。