这个系列写了好几篇文章,这是相关文章的索引,仅供参考:
以下是相关深度学习工具包的安装,包括Tensorflow, PyTorch, Torch等:
首先安装libcupti-dev
sudo apt-get install libcupti-dev
然后用 virtualenv 方式安装 Tensorflow(当前是1.4版本)
sudo apt-get install python-pip python-dev python-virtualenv
mkdir tensorflow
cd tensorflow
virtualenv --system-site-packages venv
source venv/bin/activate
pip install --upgrade tensorflow-gpu
测试GPU:
Python 2.7.12 (default, Nov 19 2016, 06:48:10)
[GCC 5.4.0 20160609] on linux2
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
...
2017-10-24 20:37:24.290049: I tensorflow/core/common_r