服务器 linux anaconda 安装GPU版本tensorflow,keras，pytorch

Catherine_xy

于 2018-12-10 11:28:36 发布

阅读量601

点赞数

CC 4.0 BY-SA版权

分类专栏： tensorflow

1 篇文章

订阅专栏

本文详细介绍如何使用conda创建Python 3.6环境，并安装TensorFlow GPU版、Keras、PyTorch及 torchvision等深度学习框架。文章还提供了安装验证方法，确保安装过程无误。

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

1.建环境

conda create --name ly1 python=3.6
source activate ly1  #激活环境
source deactivate  #退出环境

2.安装GPU版本tensorflow

pip install tensorflow-gpu=1.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

3.安装keras

pip install keras

4.安装pytorch（torchvision）

pip install torch-0.2.0.post3-cp36-cp36m-manylinux1_x86_64.whl

pip install torchvision

安装完成后，测试安装是否成功

python
import tensorflow
tensorflow.__version__#查看tensorflow版本