tensorflow-gpu==1.15 cuda10.1 pytorch 1.7.0 坑与解决方法

坑

一开始先安装的cuda10.1

然后安装pytorch 1.7.0，

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.1 -c pytorch

最后装tensorflow的时候，pip install tensorflow-gpu==1.15完成后，会报***_100.dll 找不到。参考网上的解决方案，将***101.dll改成***100.dll , 6个相关文件全部改完，代码还是跑不起来。

解决方案

以上问题是因为安装pytorch时候有cudatoolkit=10.1导致的。所以先安装tensorflow 和cudatoolkit=10.0，然后再安装pytorch，就可以了。

pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html