UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error?

看到这个问题肯定是我们的cuda 找不到,这时候你使用nvcc -V 查看版本号发现什么没有。

处理问题:

首先看以下你是否安装类cuda:  ls /usr/local/   

如果没有安装了安装,如果安装了 你需要去看一下~/.bashrc文件并把下面命令加入其中

export CUDA_HOME=/usr/local/cuda/bin

最后使~/.bashrc文件生效 :source ~/.bashrc

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐