博客
关于我
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/

    你可能感兴趣的文章
    npm install 报错 fatal: unable to connect to github.com 的解决方法
    查看>>
    npm install 报错 no such file or directory 的解决方法
    查看>>
    npm install 权限问题
    查看>>
    npm install报错,证书验证失败unable to get local issuer certificate
    查看>>
    npm install无法生成node_modules的解决方法
    查看>>
    npm install的--save和--save-dev使用说明
    查看>>
    npm node pm2相关问题
    查看>>
    npm run build 失败Compiler server unexpectedly exited with code: null and signal: SIGBUS
    查看>>
    npm run build报Cannot find module错误的解决方法
    查看>>
    npm run build部署到云服务器中的Nginx(图文配置)
    查看>>
    npm run dev 和npm dev、npm run start和npm start、npm run serve和npm serve等的区别
    查看>>
    npm run dev 报错PS ‘vite‘ 不是内部或外部命令,也不是可运行的程序或批处理文件。
    查看>>
    npm scripts 使用指南
    查看>>
    npm should be run outside of the node repl, in your normal shell
    查看>>
    npm start运行了什么
    查看>>
    npm WARN deprecated core-js@2.6.12 core-js@<3.3 is no longer maintained and not recommended for usa
    查看>>
    npm 下载依赖慢的解决方案(亲测有效)
    查看>>
    npm 安装依赖过程中报错:Error: Can‘t find Python executable “python“, you can set the PYTHON env variable
    查看>>
    npm.taobao.org 淘宝 npm 镜像证书过期?这样解决!
    查看>>
    npm—小记
    查看>>