原文:pytorch查看CUDA支持情况,只需要三行代码,另附Cuda runtime error (48) : no kernel image is available for execution处理办法

上面的命令只是检测CUDA是否安装正确并能被Pytorch检测到,并没有说明是否能正常使用,要想看Pytorch能不能调用cuda加速,还需要简单的测试一下: 一般来讲,输出主要是报 号错误,也就是CUDA的问题,出现这个问题在于硬件的支持情况,对于算力 . 的显卡来说,如果安装了 . 的CUDA就会出现这个问题,解决的办法是退回CUDA . ,或者更换更加高端的显卡,或者直接从源码编译,并在源 ...

2018-12-20 15:14 3 10877 推荐指数:

查看详情

Cuda runtime error (999)

本人使用的是Ubuntu18.04,使用GPU进行深度炼丹的时候出现了Cuda runtime error(999)的错误。 参考:https://discuss.pytorch.org/t/cuda-runtime-error-999/69658/11 “ Ubuntu seems ...

Mon Aug 10 18:34:00 CST 2020 0 935
 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM