在线安装cuda tensorflow

本文详细介绍如何在Ubuntu系统上安装CUDA 8.0、cuDNN 6.0及TensorFlow 1.3,包括配置环境变量及安装依赖软件等步骤。

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

首先,下载需要安装的包 cuda(8.0),cudnn(6.0),tensorflow-1.3

将所有文件放入主目录下的一个文件夹,我放在~/fk/111

然后执行下面这个.sh 文件就可以了

.sh文件的内容如下:

#!/bin/bash
#sudo tar xzf ~/fk.tar.gz
cd ~/fk/111/
sudo apt-get install -y gcc g++ gfortran build-essential git wget linux-image-generic libopenblas-dev python-dev python-pip python-nose python-numpy python-scipy Python-Opencv
sudo dpkg -i cuda-repo-ubuntu1404-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt-get update
sudo chmod +777 -R /opt/
sudo chmod +777 -R /data/
sudo mkdir  ~/.pip/
echo "[global]
timeout = 60
index-url = http://pypi.douban.com/simple
trusted-host = pypi.douban.com" >>~/.pip/pip.conf
sudo apt-get install cuda
#sudo cp -a profile /etc/profile
source /etc/profile
tar xzf cudnn-8.0-linux-x64-v6.0.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo pip install -U pip
sudo pip install ~/fk/111/tensorflow_gpu-1.3.0-cp27-cp27mu-manylinux1_x86_64.whl
source /etc/profile
#sudo rm -fr cuda
#sudo tar xf cudnn-7.5-linux-x64-v5.1.tar
#sudo cp cuda/include/cudnn.h /usr/local/cuda/include
#sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64


由于我是批量安装,所有其中的profile文件是我在原来的文件后面加了环境变量

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH

你可以在开始先对profile进行操作,然后省却sudo cp -a profile /etc/profile 这条命令

后面三行是我又将cudnn6.0换成5.0,不会影像tensorflow的使用

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值