上述代码将基于nvidia/cuda:11.4.0-runtime-ubuntu20.04上创建Docker镜像,并安装了cmake和构建工具build-essential,同时也清理了APT缓存以减少镜像的体积。 添加CUDA支持到CMakeLists.txt 接下来,我们需要创建一个简单的CUDA示例项目,并在CMakeLists.txt文件中添加CUDA支持。首先创建项目目录模型如下: cuda_project/ ├...
首先,打开tensorflow_cc/cmake/TensorflowBase.cmake,在第23行(COMMAND tensorflow/contrib/makefile/download_dependencies.sh之前)加入: COMMAND sed -i "s#EIGEN_URL=.*#EIGEN_URL=\"https://gitlab.com/libeigen/eigen/-/archive/3.3.7/eigen-3.3.7.tar.gz\"#g" tensorflow/contrib/makefile/download_de...
51CTO博客已为您找到关于docker环境中cmakelist 添加cuda的相关内容,包含IT学习相关文档代码介绍、相关教程视频课程,以及docker环境中cmakelist 添加cuda问答内容。更多docker环境中cmakelist 添加cuda相关解答可以来51CTO博客参与分享和学习,帮助广大IT技术人实现成长和进