查看是否可以安装GPU环境
第一步:判断能否按照gpu版本,如果是英伟达的显卡,就可以安装,如果不是英伟达的就不可以安装gpu版本,只可以安装cpu版本。
- 判断自己的电脑是否含有英伟达显卡,右键点击任务栏空白处,选择打开任务管理器。
- 打开性能选项,找到并点击GPU,有的电脑有2块GPU,有的只有一块,我的电脑是2块,打开后看右上角,我的电脑是英伟达的,只要有一块是英伟达的就可以安装GPU版本。
查看安装哪儿个版本的CUDA及CUDNN
win+R打开终端,输入nvidia-smi命令,下面是我的电脑的配置信息,红框为电脑所能支持的最大的CUDA版本,我们所安装的版本只需要比他低就可以
在终端输入nvcc -V就可以查看安装的cuda版本。
还需要下载CUDNN,需要与CUDA配套,下载好的CUDNN需要复制到你下载CUDA的目录里面去,一般是C盘programfiles里面NVIDIA GPU Computing Tookit里面的CUDA里面的V11.1里面去。