以下是我自己摸索后的经验方法,如有不对,欢迎讨论。
第一种方法:安装软件法
1.先安装Anaconda3
2.然后再安装cuda和cudnn(这两个的版本号必须是对应关系)
安装完成后,把cuda文件夹拷贝到cuda安装路径里面的cuda文件夹。(这步操作就是安装cudnn)
#############################################################################
第二种方法:安装在虚拟环境中
这个方法必须先安装显卡驱动(NVIDIA)----最好是最新的驱动
再在安装Anaconda3(和第一种方法安装一样)
然后在Anaconda3中创建虚拟环境,在虚拟环境中安装cuda和cudnn。(直接输入安装命令即可)
# 创建一个名字为env_name的虚拟环境,python版本为3.8(可以指定其他版本)
conda create -n env_name python=3.8
# 激活虚拟环境
activate env_name
# 在当前虚拟环境中安装cuda和cudnn(可以指定其他版本号)
# 这里的cudatoolkit就是cuda
conda install cudatoolkit=10.1 cudnn=7.6.5
---------------------------说明------------------------------
无论用哪个方法安装完Anaconda3,cuda,cudnn后,都需要找到cuda对应可以使用的的tensorflow和pytorch的版