1.命令行:nvidia-smi查看显卡驱动确定cuda最高支持的版本(如12.3,向下兼容)
2.根据cuda版本去英伟达官网下载cuda安装包
3.为cuda添加环境变量
4.安装支持cuda版本的cudnn
5.将cudnn拖动到对应的cuda目录下
6.pytorch官网安装与cuda对应版本的pytorch
pytorch安装
最新推荐文章于 2024-07-24 19:41:07 发布
1.命令行:nvidia-smi查看显卡驱动确定cuda最高支持的版本(如12.3,向下兼容)
2.根据cuda版本去英伟达官网下载cuda安装包
3.为cuda添加环境变量
4.安装支持cuda版本的cudnn
5.将cudnn拖动到对应的cuda目录下
6.pytorch官网安装与cuda对应版本的pytorch