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版本一定是低于本机显卡支持的版本



原文地址:https://www.cnblogs.com/bianfuxia/p/15630325.html