win10配置MMClassification+PyTorch+CUDA

Win10配置MMClassification

依赖

  1. Python 3.8
  2. CUDA 10.2
  3. Microsoft Visual C++ 14.0
  4. PyTorch 1.10.0
  5. MMCV 1.3.17
  6. MMClassification 0.17.0

配置CUDA

安装CUDA Toolkit

  1. 根据需要,下载CUDA Toolkit 10.2,选择本地安装

    image-20211122135450079

  2. 双击exe文件,安装到默认路径下

  3. 选择自定义安装,除CUDA/Visual Studio Integration外全部选择安装

  4. 查看系统环境变量是否已经自动配置

  5. 测试是否安装成功

    1. nvcc --version

    2. set cuda

### 安装支持 CUDA 12.1 的 PyTorch 为了在 Windows 11 上成功安装并配置支持 CUDA 12.1 的 PyTorch,以下是详细的说明: #### 系统需求确认 确保 NVIDIA 显卡驱动已更新至最新版本,并通过 NVIDIA 控制面板验证当前系统的 CUDA 版本。可以通过以下方式检查显卡的 CUDA 支持情况: - 打开 NVIDIA 控制面板。 - 转到左侧菜单中的“系统信息”选项。 - 在“组件”标签下查看 CUDA 的具体版本号[^1]。 如果显示的 CUDA 版本为 `12.1` 或更高,则可以继续下一步操作;否则需先升级 GPU 驱动程序以满足最低要求。 #### 下载必要的库文件 对于完整的 CUDA 和 cuDNN 功能实现,建议从官方或其他可信资源获取对应的二进制包。例如,在某些情况下可能需要手动下载 CUDA 工具链以及配套的 cuDNN 库[^2]: ```bash # 使用百度网盘链接作为替代方案之一 (注意密码保护机制) wget https://pan.baidu.com/s/1aOkHVXP2EuwMFYZx0oS9HA -O cudnn_cuda.zip && unzip cudnn_cuda.zip ``` > **重要提示**: 如果上述方法不可用或者存在安全顾虑, 推荐访问[NVIDIA官网](https://developer.nvidia.com/cuda-downloads)寻找正式发行版. #### Conda 创建虚拟环境与依赖管理 推荐采用 Anaconda 来简化 Python 及其科学计算生态的部署流程。执行如下命令创建一个新的 conda 环境,并指定所需的 PyTorch 组件及其兼容性设置: ```bash # 初始化新的 conda env 并激活它 conda create --name torch-env python=3.9 conda activate torch-env # 添加 pytorch 渠道源 (-c 参数),同时指明目标 cuda 版本(pytorch-cuda=...) conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia ``` 尽管这里选择了较旧但稳定的组合(`pytorch-cuda=11.8`)来匹配大部分硬件条件[^3], 用户也可以尝试调整参数直至完全适配最新的 cuda12.x 构建成果。 完成以上步骤之后,请运行简单的测试脚本来检验整个框架能否正常调用GPU加速功能: ```python import torch print(torch.cuda.is_available()) # 输出 True 表示可用 print(torch.version.cuda) # 展现实际加载的 cuda driver info ``` --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值