当前位置: 首页 > 工具软件 > Tensor2Tensor > 使用案例 >

tensor2tensor在GTX1650上的安装

金阳曜
2023-12-01

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
  1. 安装cuda和cudnn
conda install cudnn=7.6.0 cudatoolkit=10.0
  1. 安装t2t
pip install tensor2tensor==1.14.1
  1. 将上一步安装的tensorflow换为1.15.0
pip uninstall tensorflow
pip install tensorflow==1.15.0
  1. 正常来说此时已经安装完成,但会出现tensorflow-datasets版本不匹配的情况,安装tensorflow-datasets==3.2.1:
pip uninstall tensorflow-datasets
pip install tensorflow-datasets==3.2.1
 类似资料: