经验首页 前端设计 程序设计 Java相关 移动开发 数据库/运维 软件/图像 大数据/云计算 其他经验
当前位置:技术经验 » 大数据/云/AI » 人工智能基础 » 查看文章
深度学习GPU加速配置方法
来源:cnblogs  作者:Mr·Jiang  时间:2022/12/12 18:27:12  对本文有异议

深度学习GPU加速配置方法

一、英伟达官方驱动及工具安装
  1. 首先检查自己的电脑驱动版本,未更新至最新建议先将驱动更新至最新,然后点击Nvidia控制面板

2.在如下界面中点击系统信息,点击显示可以看见当前的显卡驱动版本,点击组件可以看到红框中的CUDA版本的最高支持,在安装时只需要装这个版本之下的即可。

  1. 打开Cuda Tookit的安装官网,CUDA Toolkit Archive | NVIDIA Developer,选择自己需要的版本,但不能高于上面的版本号,此处以Cuda Tookit10.2为例,选择好对应的版本号和系统,进行下载。
  1. 选择自定义安装,注意在安装这之前至少得有一个Visual Studio,不然会安装失败。

  2. 在自定义中如果是第一次安装则选择所有的选项,如果是第n次安装,则只选择第一个选项,安装位置尽量默认。

  3. 安装完成后如图所示:

  4. 然后在命令行中输入nvcc -V即可查看安装的cuda。

    PS:如果未出现考虑环境变量的问题,主要参考以下几个环境变量,注意安装位置是默认位置

  5. 安装完cuda后,还需安装cuda的补丁即cuDnn,下载网址cuDNN Download | NVIDIA Developer,因为这里安装的是10.2版本的cuda,所以选择8.7版本的cuDnn for Cuda 10.2,然后根据自己的平台选择相应的版本,这里选择windows压缩包。

  6. 将下载的文件进行解压,将这三个文件进行复制,然后放入安装目录(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)下的同样文件。

  7. 此时可以打开pytorch官网,此处以pytorch V1.11.0为例,需要安装对应cuda版本的pytorch,此处为10.2版本,复制安装命令即可。

  1. 注意此处必须是cuda版本,有时候利用换源安装的是cpu版本,注意区分

  2. 输入如下命令,如果最后结果显示True,说明安装成功,恭喜可以进行GPU加速了

原文链接:https://www.cnblogs.com/yyjn/p/16976843.html

 友情链接:直通硅谷  点职佳  北美留学生论坛

本站QQ群:前端 618073944 | Java 606181507 | Python 626812652 | C/C++ 612253063 | 微信 634508462 | 苹果 692586424 | C#/.net 182808419 | PHP 305140648 | 运维 608723728

W3xue 的所有内容仅供测试,对任何法律问题及风险不承担任何责任。通过使用本站内容随之而来的风险与本站无关。
关于我们  |  意见建议  |  捐助我们  |  报错有奖  |  广告合作、友情链接(目前9元/月)请联系QQ:27243702 沸活量
皖ICP备17017327号-2 皖公网安备34020702000426号