1. 问题原因
代码需要使用libcudart.so.10.2,但是当前cuda版本不是10.2,所以缺少该文件。
2. 解决方案
第一步:
- 在cuda官网下载10.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