美文网首页
PyTorch的使用-----1. CUDA安装

PyTorch的使用-----1. CUDA安装

作者: wendy云泽 | 来源:发表于2020-04-02 20:40 被阅读0次

    1. 下载cuda

    网址:https://developer.nvidia.com/cuda-downloads,选择win10版本

    cuda下载页面

    2.安装

    安装路径自定义,这里我是:D:\Users\eleven\AppData\Local\Temp\CUDA

    安装的时候建议选择自定义而不是“精简”(从下面的英文解释可以看出,其实这里的精简写成完整应该更贴切,他会安装所有组件并覆盖现有驱动,然而我并不想安装全家桶,何况我的官方显卡驱动比他的新)。

    其余直接next

    如下图所示,我们只需选择CUDA项(默认是全选的),这一步之后,会询问这些组件的安装路径,可以直接使用C盘的默认位置,当然我们可以可以自己定义(请记住这些安装路径,后面配置环境变量需要用到)。

    3. 检查环境变量

    添加三个环境变量:

    检查是否成功:

    显示cuda版本号10.2  V10.2.89,说明安装成功!

    相关文章

      网友评论

          本文标题:PyTorch的使用-----1. CUDA安装

          本文链接:https://www.haomeiwen.com/subject/xtcfphtx.html