1. 问题原因

代码需要使用libcudart.so.10.2,但是当前cuda版本不是10.2,所以缺少该文件。

2. 解决方案

第一步:
  1. 在cuda官网下载10.2版本,
  2. 将/usr/local/cuda-10.2/lib64下的libcudart.so.10.2复制到当前系统版本对应的cuda/lib64文件夹下
    cp /usr/local/cuda-10.2/lib64/libcudart.so.10.2 /usr/local/cuda-xxx/lib64/
    
    使用nvcc --version命令查看当前系统变量中设置的cuda版本
第二步:

更新和配置动态链接器的缓存

sudo ldconfig /usr/local/cuda-xxx/lib64

done

Logo

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

更多推荐