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