Linux上tensorflow环境搭建
安装pip
sudo apt install python3-pip
pip install tensorflow说明:tensorflow 2.5.0有bug,可以安装测试版
pip install tf-nightly安装CUDA 10.0
sudo sh cuda_11.0.2_450.51.05_linux.run{{/code}}
CUDA 11.3下载地址:
{{code language="none"}}
wget https://developer.download.nvidia.com/compute/cuda/11.3.1/local_installers/cuda_11.3.1_465.19.01_linux.run创建缺少的链接符号`libcusolver.so.11`(这是tensorflow 2.5.0 bug引起的)
sudo ln -s /usr/local/cuda-11.0/lib64/libcusolver.so.10.5.0.218 /usr/local/cuda-11.0/lib64/libcusolver.so.11安装CUDNN 8
最后更新于1年前
本文由人工编写,AI优化,转载请注明原文地址: Linux上tensorflow环境搭建
推荐阅读
评论 (0)
发表评论
昵称:加载中...
暂无评论,快来发表第一条评论吧!