关于配置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的卷积神经网络训练