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

本文共 2221 字,大约阅读时间需要 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.whlpip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
  • 方法二:使用 .whl 文件安装

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

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

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

  • 五、验证配置

  • 验证 PyTorch 版本

    import torchprint(torch.__version__)

    预期输出:torch 1.2.0

  • 验证 CUDA 加速

    import torchx = 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/

    你可能感兴趣的文章
    Plotly 绘制表面 3D 未显示
    查看>>
    Plotly-Dash 存在未知问题并创建“加载依赖项时出错“;通过使用 Python-pandas.date_range
    查看>>
    Plotly-Dash:如何过滤具有多个数据框列的仪表板?
    查看>>
    Plotly:如何为 x 轴上的时间序列设置主要刻度线/网格线的值?
    查看>>
    Plotly:如何从 x 轴删除空日期?
    查看>>
    Plotly:如何从单条迹线制作堆积条形图?
    查看>>
    Plotly:如何以 Root 样式绘制直方图,仅显示直方图的轮廓?
    查看>>
    Plotly:如何使用 Plotly Express 组合散点图和线图?
    查看>>
    Plotly:如何使用 plotly.graph_objects 和 plotly.express 定义图形中的颜色?
    查看>>
    Plotly:如何使用 Python 对绘图对象条形图进行颜色编码?
    查看>>
    Plotly:如何使用 updatemenus 更新一个特定的跟踪?
    查看>>
    Plotly:如何使用长格式或宽格式的 pandas 数据框制作线图?
    查看>>
    Plotly:如何向烛台图添加交易量
    查看>>
    Plotly:如何在 plotly express 中找到趋势线的系数?
    查看>>
    Plotly:如何在桑基图中设置节点位置?
    查看>>
    pm2 start命令中的json格式详解
    查看>>
    pm2启动报错
    查看>>
    pm2通过配置文件部署nodejs代码到服务器
    查看>>
    PML调用PDMS内核命令研究
    查看>>
    PMM安装-第一篇
    查看>>