tensor2tensor在GTX1650上的安装

tensor2tensor在GTX1650上的安装

如题，在做一个NLP类型的项目，需要用到google公司的tensor2tensor，简称t2t（弱弱吐槽一下t2t的环境有点难搞），于是在自己的笔记本上先尝试了一下t2t的安装

硬件环境

NVIDIA GeForce GTX1650

软件环境

1. Anaconda（我用的是4.8.3，其实conda版本没多大要求）
2. python3.6（3.7总是报NoneType:Attribute don’t have ‘copy’）
3. cuda10.0
4. cudnn7.6.x（7.6.0或7.6.5都可以）

虽然conda不是必需，但还是推荐装一个conda，因为conda可以通过如下命令：

conda install cudnn=7.6.0 cudatoolkit=10.0

在conda环境中安装cuda和cudnn，与本机的cuda不冲突。
否则本机需要更换cuda，而此时又有可能影响了其他项目，尤其是依靠本机cuda才能运行的项目。
因此推荐使用conda安装。

安装过程

（1，2步如果没有安装conda可跳过，不影响pip安装）

1. 创建一个python3.6的conda环境

conda create -n 环境名字 python=3.6

2. 安装cuda和cudnn

conda install cudnn=7.6.0 cudatoolkit=10.0

3. 安装t2t

pip install tensor2tensor==1.14.1

4. 将上一步安装的tensorflow换为1.15.0

pip uninstall tensorflow
pip install tensorflow==1.15.0

5. 正常来说此时已经安装完成，但会出现tensorflow-datasets版本不匹配的情况，安装tensorflow-datasets==3.2.1：

pip uninstall tensorflow-datasets
pip install tensorflow-datasets==3.2.1