Win10安装NVIDIA驱动、CUDA、CUDNN、PyTorch
- 1 NVIDIA驱动下载
- 2 下载CUDA
- 2.1 查看CUDA版本
- 2.2 下载并安装CUDA
- 3 下载并安装CUDNN
- 3.1 输入账号和密码后，登录CUDNN官方网站进行下载
- 3.2 安装
- 3.2.1 解压下载的压缩包
- 3.2.2 将解压后bin目录的内容全部放到CUDA对应的bin目录
- 3.2.3 将解压后include目录的内容全部放到CUDA对应的include目录
- 3.2.4 将解压后lib目录的内容全部放到CUDA对应的lib目录
- 3.2.5 打开系统环境变量，编辑Path变量，添加CUDA路径
- 3.2.6 验证
- 4 安装PyTorch
- 5 torch.cuda.is_available()==false
1 NVIDIA驱动下载
NVIDIA驱动下载链接
（1）选择合适的型号，然后搜索
（2）点击下载
（3）同意协议并下载
（4）下载完成后，安装的步骤按正常流程走，此处省略安装步骤。
2 下载CUDA
2.1 查看CUDA版本
（1）点击NVIDIA控制面板，NVIDIA Control Panel
（2）选中“管理3D设置”，点击下方的点击下方的“系统信息”，点击“组件”按钮，查看CUDA版本；我这边是12.2
”
2.2 下载并安装CUDA
1、进入CUDA官网，找到对应版本
安装CUDA后，使用在终端中验证：
使用nvidia-smi或者nvcc --version
3 下载并安装CUDNN
CUDNN官网
3.1 输入账号和密码后，登录CUDNN官方网站进行下载
3.2 安装
3.2.1 解压下载的压缩包
3.2.2 将解压后bin目录的内容全部放到CUDA对应的bin目录
3.2.3 将解压后include目录的内容全部放到CUDA对应的include目录
3.2.4 将解压后lib目录的内容全部放到CUDA对应的lib目录
3.2.5 打开系统环境变量，编辑Path变量，添加CUDA路径
将这四条路径添加进path，前两条在之前已经自动添加
3.2.6 验证
4 安装PyTorch
在win10系统上安装PyTorch，可直接通过Torch官网的命令进行下载。
官网链接：https://pytorch/get-started/locally/
5 torch.cuda.is_available()==false
如果发现安装的是cpu版本的torch，不要紧，下载pytorch-2.0.1-py3.8_cuda11.7_cudnn8_0：
文件下载网站：https://mirrors.tuna.tsinghua.edu/anaconda/cloud/pytorch/win-64/
运行安装命令：
conda install --offline pytorch-2.0.1-py3.8_cuda11.7_cudnn8_0.tar.bz2
再在终端的python环境中，输入torch.cuda.is_available() 验证，返回True .大功告成！
发布者：admin，转转请注明出处：http://www.yc00.com/web/1755056170a5233960.html
评论列表（0条）