安装tensorflow gpu enable python 2.7 版本

来源:互联网 发布:mac终端抹掉磁盘命令 编辑:程序博客网 时间:2024/06/05 11:14
只需要输入如下代码:export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whlsudo pip install --upgrade $TF_BINARY_URL2.tensorflow官网上要求的环境变量:
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"export CUDA_HOME=/usr/local/cuda

0 0
原创粉丝点击