网站开发前台/新媒体运营怎么自学
服务器跑模型出现Kernel not compiled with GPU support
1、环境
一般我们在跑模型的时候,需要搞清楚该模型所需要的环境信息,建立一个虚拟环境,根据模型所需的环境进行安装
一般我们喜欢使用Anaconda 来建立虚拟环境,若没安装的可以查看另外一篇博文
Linux 安装 Anaconda
2、问题描述
当我们跑模型/安装包的时候出现,RuntimeError: Not compiled with GPU support,一般这种就是你模型所需cuda版本信息与你本机的cuda版本信息不一致导致的
3、问题解决
关于显卡、显卡驱动、CUDA、CUDAnn 等不懂知识可以查看另外一篇博文
显卡,显卡驱动,nvcc, cuda driver,cudatoolkit,cudnn
(1)首先要确定模型所需的环境信息。
例如Pytorch的版本 找到对应的cuda的版本信息,进行下载,详情可以看上面的那个链接
CUDA下载链接
(2)运行nvcc --version查看你的 cuda 编译器版本,那么你的 pytorch-gpu 也建议安装对应版本。当然如果你nvcc都没安装。。。那你就先找教程安装。
(3)如果安装的 pytorch 版本和nvcc版本一致,你可以看一下你的 CUDA 路径是否在~/.bashrc中设置正确,参考的配置路径如下:
# vim ~/.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:/usr/local/cuda/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/lib
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64
设置好后 source ~/.bashrc
然后删除模型对应的build文件夹(如果你之前已经编译过一遍了),然后重新编译一下模型