1.确认存在cuda和cudnn。
2.pip安装cuda对应的版本tf-gpu,
3.然后可能会出现如下报错:
Could not load dynamic library ‘cusolver64_10.dll’; dlerror: cusolver64_10.dll not found
解决方式:
conda install cudatoolkit=11.0
4.tensorflow-gpu=2.4.0对应transformers=4.1
一般采用tensorflow -> matplotlib -> seaborn -> skleran 的顺序,就不需要另外安装pandas和numpy的依赖库,因为都在上面的安装过程中安装好了