PyTorch代码迁移至TensorFlow的可行性探讨

1、关于深度学习框架的安装，如果单纯安装cpu版本的直接去下载到本地即可，用pip安装

torch网址  https://download.pytorch.org/whl/torch_stable.html

tenorflow地址

tensorflow-gpu下载地址：http://mirrors.aliyun.com/pypi/simple/tensorflow-gpu/
tensorflow-cpu下载地址：https://pypi.org/project/tensorflow/#files

2、安装cpu没有啥注意的，主要是gpu版本的，首先要下载cuda，然后下载cudnn,同时要看对应支持的版本，下载的cuda版本必须比显卡支持的版本要低，否则是不行的

下载的cuda版本，和cudnn对应的版本都要一样。cudnn下载好进行解压，然后把bin,include,lib对应的文件复制到cuda下bin,include,lib文件夹下，记住不是覆盖，而是挪过去。

然后装对应cuda版本对应的torch版本  一定要符合

3、在装的过程中可能需要对驱动程序进行升级。装的cuda版本一定是低于本机显卡支持的版本

免责声明：本文系网络转载或改编，未找到原创作者，版权归原作者所有。如涉及版权，请联系删