我回忆一下,是先安装好Tensorflow GPU,运行了一些测试,比如分词等。然后新建虚拟环境安装PyTorch GPU,再运行一些测试,感觉PyTorch加载模型要快一些。然后安装了一个两者都有的环境,模型加载的速度比Tensorflow环境要快,所以觉得是优先加载PyTorch。
文档中没有说明,问一下确认一下。我笔记本上的组合是NVIDIA GeForce RTX 2060 with Max-Q + Driver 27.21.14.6589 + cuda 11.3 + cudnn 8.2.1 + Tensorflow 2.6 + PyTorch 1.11.0。