笔记-分享
首先要确定安装的版本--不同的tensorflow-gpu、pytorch-gpu等版本都需要对应的cudnn和cuda
可以去对应的官网查看(找到官网提供的安装命令,解析出对应的各种版本,按部就班的安装即可)
比如:安装pytorch以及对应的cudnn、cuda(tensorflow同理)
打开pytorch官网 下拉到这个位置
点击 install previous versions of PyTorch
找到你需要的pytorch-gpu的版本,如v1.11.0
那么你就需要安装cuda10.2 或者11.3才能使用这个版本的pytorch
cuda同时也需要对应的cudnn,如下
cuda10.2和cudnn7.6.5对应
安装cudnn7.6.5