关于配置tensorflow-gpu与tensorflow probability

关于配置tensorflow-gpu,以及cuda和cudnn的匹配,见:

https://blog.csdn.net/weixin_44684139/article/details/88387996

但是由于我是RTX2070显卡,可能借鉴意义不是特别大,总而言之就是:
tensorflow版本——cuda版本——cudnn版本三者要对应,一个不对应都不行

这里重点讲在tensorflow-gpu中的tensorflow probability的配置:
首先安装了tensorflow-gpu 1.14,(因为装其他版本会报cudnn初始化失败,因此这里的gpu版本一定与你之前配置tensorflow-gpu的版本一致,也就是找到属于你自己电脑的版本)然后

pip install --user --upgrade tfp-nightly

即可安装tensorflow probability
然后进行一波导入

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
from tqdm import tqdm_notebook as tqdm
import urllib.request
from sklearn.model_selection import train_test_split
import tensorflow_probability as tfp

但是报错:

ImportError: cannot import name ‘type_spec’

如何解决呢,需要重新pip:

pip install tf-nightly-2.0-preview

完美解决所有问题,即不报cudnn初始化失败的错,又不报导入包时候的错

接下来训练——完美完成tfp的卷积神经网络训练

  • 2
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值