博客
关于我
Windows10 + cuda10.0 + pytorch1.2.0 + torchvision0.4.0环境配置
阅读量:595 次
发布时间:2019-03-12

本文共 2227 字,大约阅读时间需要 7 分钟。

配置 Windows 10 系统中的 CUDA 10.0、PyTorch 1.2.0 和 TensorVision 0.4.0 开发环境

这篇文档详细提供了在 Windows 10 系统上搭建 CUDA 10.0、PyTorch 1.2.0 和 TensorVision 0.4.0 开发环境的步骤说明。对于深度学习涉及图形处理任务的开发者来说,这是一个常用的配置,以下内容将引导你顺利完成环境搭建。


一、准备工作

在开始配置前,请确保你的系统满足以下条件:

  • 操作系统:Windows 10 Professional 或更高版本。
  • 显卡驱动:NVIDIA 显卡,建议使用支持 CUDA 的 RTX 系列显卡。
  • 硬盘空间:至少 20 GB 的可用空间,用于安装相关软件工具。

二、安装 Anaconda Python 环境

PyTorch 和相关工具的安装需要使用 Python 环境。在这里,推荐使用 Anaconda 可容东西包管理器,这样可以方便地管理开发环境。

  • 访问 Anaconda 官方网站:前往 Anaconda 官网 下载安装文件。

  • 下载并安装 Anaconda

    • 下载完成后双击运行 Anaconda 安装文件。
    • 遵循安装向导,一路选择“继续”和“同意”,直到完成安装。
  • 设置 Anaconda 环境: -安装完成后,打开 Anaconda 存储库DLL 文件,打开 Command Prompt,输入 conda create -n pytorch anaconda-python 3.8

    • 按照提示输入_conda 密码,并等待环境创建完成。
  • 激活自定义环境

    • 输入命令 conda activate pytorch,打开 Anaconda 环境。

  • 三、安装 CUDA 10.0 和 cuDNN

    CUDA 是图形处理的加速库,用于 PyTorch 的加速。在此版本中,我们会安装 CUDA 10.0 和相应的 cuDNN。

  • 访问 NVIDIA 官方下载页面:前往 CUDA 10.0 下载页面

  • 选择合适的 CUDA 版本

    • 在页面顶部选择目标操作系统(Windows),然后根据显卡系列选择相应的 CUDA 版本。
    • 这里选择 CUDA 10.0 最新版本,并点击开始下载 CUDA 安装文件。
  • 安装 CUDA

    • 双击下载的 CUDA .exe 文件,遵循安装向导,勾选“自定义安装”选项,设置安装路径、选择要安装的内容,然后点击“开始”。
  • 安装 cuDNN

    • 访问 cuDNN 下载页面
    • 登录 NVIDIA 账号,下载对应 CUDA 版本的 cuDNN 启用版,解压后找到下载文件。
  • 配置 cuDNN 环境

    • 将 cuDNN 的三个文件夹 (cudnn、include、lib) 拷贝到 CUDA 安装路径下的对应位置。
    • 验证安装:在命令提示符中输入 nvcc -V,确认 CUDA 版本是否正确安装。

  • 四、安装 PyTorch 1.2.0 和 TensorVision 0.4.0

    方法一:使用 pip 安装

  • 访问 PyTorch 官方下载页面:前往 PyTorch 官网下载页面

  • 下载 PyTorch 和 TensorVision

    • 单独下载 .whl 文件(例如 torch-1.2.0-cp36-cp36m-win_amd64.whltorchvision-0.4.0-cp36-cp36m-win_amd64.whl),确保版本与 CUDA 和 cuDNN 一致。
  • 安装依赖项:使用 pip 命令安装:

    pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
    pip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
  • 方法二:使用 .whl 文件安装

  • 在已有环境中打开命令提示符

  • 根据需要输入以下命令,确保下载路径正确:

    pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
    pip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl

  • 五、验证配置

  • 验证 PyTorch 版本

    import torch
    print(torch.__version__)

    预期输出:torch 1.2.0

  • 验证 CUDA 加速

    import torch
    x = torch.randn(3, 3)
    print(x_folder.device) # 检查设备是否为 CUDA

  • 六、注意事项

  • 环境变量路径

    • 确保 PATH 环境变量中添加了 CUDA 的安装目录。
    • 打开 environment variables,手动添加路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\10.0\bin
  • CuDNN 版本匹配

    • 确保下载的 cuDNN 版本与 CUDA 版本一致,避免兼容性问题。
  • 系统重启

    • 安装完成后建议重启系统,以确保所有 CUDA 相关服务正常运行。

  • 通过以上步骤,你已经成功搭建了 CUDA 10.0、PyTorch 1.2.0 和 TensorVision 0.4.0 的开发环境。如果在安装过程中遇到问题,可以参考 PyTorch 官方文档 进行进一步的求助。

    转载地址:http://qlhtz.baihongyu.com/

    你可能感兴趣的文章
    NOIp模拟赛二十九
    查看>>
    Vue3+element plus+sortablejs实现table列表拖拽
    查看>>
    Nokia5233手机和我装的几个symbian V5手机软件
    查看>>
    non linear processor
    查看>>
    Non-final field ‘code‘ in enum StateEnum‘
    查看>>
    none 和 host 网络的适用场景 - 每天5分钟玩转 Docker 容器技术(31)
    查看>>
    None还可以是函数定义可选参数的一个默认值,设置成默认值时实参在调用该函数时可以不输入与None绑定的元素...
    查看>>