一. 安装CUDA
二. 安装CUDNN
三,安装GPU版本pytorch
根据网上的教程安装即可,需要注意:
1. 安装完CUDA和CUDNN后先查看环境变量是否自己生成,尽量安装路径不要变更(据网友说这个也不会占用太多空间)。
2. 安装CUDA和CUDNN尽量低于电脑GPU一个版本,免得出错(有网友说同版本会出错)。保持pytorch版本和CUDA保持一致即可。
根据网上的教程安装即可,需要注意:
1. 安装完CUDA和CUDNN后先查看环境变量是否自己生成,尽量安装路径不要变更(据网友说这个也不会占用太多空间)。
2. 安装CUDA和CUDNN尽量低于电脑GPU一个版本,免得出错(有网友说同版本会出错)。保持pytorch版本和CUDA保持一致即可。