安装英伟达驱动 560.94、配置 CUDA12.4、安装 Conda 环境及配置 PyTorch2.5.0

安装英伟达驱动 560.94、配置 CUDA12.4、安装 Conda 环境及配置 PyTorch2.5.0

安装英伟达驱动、配置 CUDA、安装 Conda 环境及配置 PyTorch

tips:防止环境配置出错的办法就是一比一版本安装 参考链接:https://blog.csdn.net/zslefour/article/details/143441087 清华源:https://pypi.tuna.tsinghua.edu.cn/simple/torchvision/ 离线文件百度网盘:通过网盘分享的文件:nvidia_torch_env 链接: https://pan.baidu.com/s/1irBLVDIRyitgET9D5e-2YA?pwd=fqyr 提取码: fqyr

1. 安装英伟达驱动

版本

NVIDIA Driver: 560.94

安装步骤

访问 NVIDIA 官方网站,下载与你显卡兼容的驱动程序。按照安装向导完成驱动程序的安装。

验证驱动安装

打开“NVIDIA 控制面板”(右键点击桌面空白处,选择“NVIDIA 控制面板”)。在左侧菜单中选择“系统信息”,查看“驱动版本”是否为 560.94。打开命令提示符(CMD),输入以下命令并回车:nvidia-smi

如果显示 GPU 信息和驱动版本,则说明驱动安装成功。

2. 配置 CUDA 和 CUDNN

版本

CUDA: 12.4.99CUDNN: 8.9.5(与 CUDA 12.4 兼容)

安装步骤

安装 CUDA

访问 CUDA 下载页面,选择适合你操作系统的版本。按照官方指南完成 CUDA 的安装。配置 CUDA 环境变量:

在 Windows 系统中,右键点击“此电脑” -> “属性” -> “高级系统设置” -> “环境变量”。添加 CUDA_PATH 系统变量,值为 CUDA 的安装路径(例如:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4)。将 CUDA 的 bin 和 lib 文件夹路径添加到 Path 环境变量中。 安装 CUDNN

访问 CUDNN 下载页面,选择与 CUDA 12.4 兼容的版本(如 8.9.5)。下载 CUDNN 的 .zip 文件并解压。将解压后的文件夹中的内容复制到 CUDA 的安装路径中:

将 bin 文件夹中的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\bin。将 include 文件夹中的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\include。将 lib 文件夹中的文件复制到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.4\lib。

验证 CUDA 和 CUDNN 安装

验证 CUDA:

nvcc --version

如果显示 CUDA 编译器版本信息,则说明 CUDA 安装成功。

验证 CUDNN:

打开命令提示符(CMD),输入以下命令:nvidia-smi

如果显示 GPU 信息,则说明 CUDNN 安装成功。或者,运行以下 Python 代码检查 CUDNN 是否可用:import torch

print(torch.backends.cudnn.is_available()) # 应输出 True

3. 安装 Conda 环境

python版本为3.10.16

安装步骤

下载并安装 Anaconda 或 Miniconda。打开终端或命令行工具,创建一个新的 Conda 环境:conda create -n pytorch_env python=3.10

激活环境:conda activate pytorch_env

4. 安装 PyTorch 及相关包

版本

PyTorch: 2.5.0+cu124Torchvision: 0.20.0+cu124Torchaudio: 2.5.0+cu124

安装步骤

使用以下命令安装 PyTorch、Torchvision 和 Torchaudio:

pip install torch==2.5.0+cu124 torchvision==0.20.0+cu124 torchaudio==2.5.0+cu124 --extra-index-url https://download.pytorch.org/whl/cu124

如果需要手动下载 .whl 文件,可以使用以下链接:

Torch 2.5.0+cu124Torchvision 0.20.0+cu124Torchaudio 2.5.0+cu124 其他版本访问网站:https://download.pytorch.org/whl/torch/ 将文件下载到本地后,使用以下命令安装: pip install torchvision-0.20.0+cu124-cp310-cp310-win_amd64.whl

pip install torchaudio-2.5.0+cu124-cp310-cp310-win_amd64.whl

5. 验证安装

启动 Python,检查 PyTorch 是否成功安装:

import torch

print(torch.__version__) # 应输出 2.5.0

print(torch.cuda.is_available()) # 应输出 True

测试 CUDA 和 CUDNN 是否正常工作:

import torch

x = torch.randn(3, 3).cuda()

print(x)

print(torch.backends.cudnn.is_available()) # 应输出 True

以上步骤完成后,你应该已经成功安装了 NVIDIA 驱动、配置了 CUDA 和 CUDNN,创建了 Conda 环境,并安装了 PyTorch 及相关包。

相关推荐

世界杯裁判巡礼11:中超老熟人,大洋洲独苗——马修·康格
6款Linux常用远程连接工具,你最中意哪一款?
365bet体育在线备用

6款Linux常用远程连接工具,你最中意哪一款?

📅 10-04 👁️ 6540
35岁传奇!J罗明年迎第三届世界杯,曾获2014金靴+天外飞仙破门