如题,在做一个NLP类型的项目,需要用到google公司的tensor2tensor,简称t2t(弱弱吐槽一下t2t的环境有点难搞),于是在自己的笔记本上先尝试了一下t2t的安装
NVIDIA GeForce GTX1650
虽然conda不是必需,但还是推荐装一个conda,因为conda可以通过如下命令:
conda install cudnn=7.6.0 cudatoolkit=10.0
在conda环境中安装cuda和cudnn,与本机的cuda不冲突。
否则本机需要更换cuda,而此时又有可能影响了其他项目,尤其是依靠本机cuda才能运行的项目。
因此推荐使用conda安装。
(1,2步如果没有安装conda可跳过,不影响pip安装)
conda create -n 环境名字 python=3.6
conda install cudnn=7.6.0 cudatoolkit=10.0
pip install tensor2tensor==1.14.1
pip uninstall tensorflow
pip install tensorflow==1.15.0
pip uninstall tensorflow-datasets
pip install tensorflow-datasets==3.2.1