tensorflow (centos7 ,cuda9.2 cudnn7 python 3.6环境)源码安装步骤

最新推荐文章于 2024-12-06 14:42:48 发布

simple_hututu

最新推荐文章于 2024-12-06 14:42:48 发布

阅读量470

点赞数 1

CC 4.0 BY-SA版权

分类专栏： tensorflow编译安装

本文链接：https://blog.youkuaiyun.com/pwtd_huran/article/details/93777194

该文详细介绍了如何在CentOS7环境下，通过源码编译的方式安装TensorFlow，包括NCCL、Git、Python3、Bazel、Protobuf等依赖的安装步骤，以及编译和打包TensorFlow为pip可安装的whl包的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.安装nccl
&& git clone https://github.com/NVIDIA/nccl.git \
&& pip3 install cython -i https://pypi.tuna.tsinghua.edu.cn/simple \
&& cd nccl && make -j32 src.build CUDA_HOME='/usr/local/cuda-9.0/' NVCC_GENCODE="-gencode=arch=compute_61,code=sm_61" \
&& yum install -y rpm-build rpmdevtools && make -j32 pkg.redhat.build && make install \
&& cd /home && rm -rf nccl \
# 验证 nccl 是否安装成功
# cd /home && git clone https://github.com/NVIDIA/nccl-tests.git \
# cd nccl-tests && make -j32 && ./build/all_reduce_perf -b 8 -e 256M -f 2 -g 2 \
# cd /home && rm -rf nccl-tests

# cat /etc/ld.so.conf
include ld.so.conf.d/*.conf
# echo "/usr/local/lib" >> /etc/ld.so.co