Anaconda安装torch0.4.1及tensorflow

执行命令：

pip3 install tensorflow

提示错误：安装过程中h5py问题貌似没找到对应的版本 。。

解决：先安装h5py模块，载安装tensorflow

pip3 install h5py

pip3 install tensorflow

安装成功！

在pycharm中创建py文件，执行验证程序：

#!/usr/bin/env python3
# -*- coding:utf-8 -*-
# time:2019-06-10
# 测试 tensorflow1.13 和 pytorch0.4.1是否正常工作
import tensorflow as tf
import torch
from torch.backends import cudnn

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

x = torch.Tensor([1, 3])
y = torch.Tensor([2, 4])
z = x + y
# z = z.cuda()    # cpu版本没有cuda
print(z)
print(cudnn.is_acceptable(x))

 

 链接：Win10下用Anaconda3安装TensorFlow和pytorch（python3.5）