GPU版本tensorflow和pytorch安装¶
安装cuda和cudnn¶
- 安装显卡驱动网址
- 查看显卡支持的cuda最高版本!
- 查看cuda、cudnn、tensorflow版本对应关系网址
- 安装cuda下载网址，默认会添加环境变量
- 下载cudnn压缩包网址，将三个文件夹复制到cuda安装文件夹下
- ==好像不用安装cuda也行？pip安装cudatookit也可以？==
安装tensorflow-gpu¶
- 根据tensorflow版本安装
pip install tensorflow-gpu==2.x.x
安装pytorch-gpu¶
-
查看对应cuda版本的torch和torchvision，torch和torchvision版本依赖关系见网址
-
下载对应版本的torch和torchvision的whl文件网址
-
安装,先安装torch
pip install xxx.whl
参考¶
- https://blog.csdn.net/Netceor/article/details/119821270