iis服务器助手广告
返回顶部
首页 > 资讯 > 后端开发 > Python >CUDA安装教程
  • 400
分享到

CUDA安装教程

python开发语言 2023-09-02 06:09:35 400人浏览 八月长安

Python 官方文档:入门教程 => 点击学习

摘要

1 cuda的下载及安装 1.1 查看适合的cuda版本  我电脑上支持的cuda是11.6的 1.2 cuda toolkit下载 ​​​​​kCUDA Toolkit Archive | NVIDIA Developer 进入上

1 cuda的下载及安装

1.1 查看适合的cuda版本

 我电脑上支持的cuda是11.6的

1.2 cuda toolkit下载

​​​​​kCUDA Toolkit Archive | NVIDIA Developer

进入上述网页,找到适合的cuda

1.3 cuda toolkit安装

双击exe文件进行安装即可

1.4 配置环境

 打开 设置->高级系统设置->环境变量 

        红框里的是系统自动添加的,蓝框里的有些情况系统不会自动添加,需要手动添加,添加时注意自己的路径。

NVCUDASAMPLES_ROOT  C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.6NVCUDASAMPLES11_6_ROOT  C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.6

 1.4 验证

 win+R,输入cmd,输入nvcc --version查看版本号,输入set cuda查看设置的环境变量

2 cuANN下载及安装 

2.1 cuDNN下载

 下载地址如下,下载之前需要注册一下
https://developer.nvidia.com/rdp/cudnn-download

 如下链接,有适合自己的版本

cuDNN Archive | NVIDIA Developer

2.2 cuDNN配置

将cuDNN解压到D盘

 将三个文件夹拷贝到到cuda的安装目录下。默认的安装路径为

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

2.3 添加环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\binC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\includeC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libC:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp

2.4 验证

       win+R cmd进入安装目录下,再进入到 extras\demo_suite下,执行.\bandwidthTest.exe和.\deviceQuery.exe,得到下图。

3 安装GPU版的torch(不需要torch的忽略)

当跑深度学习代码时,会出现如下错误,原因是原先安装的是GPU版本的torch。

3.1 安装GPU版torch

进入官网Start Locally | PyTorch

选择各配置,复制红线链接

 

打开torch和torchvision选择适合自己的版本

 进入该路径下执行pip

pip install torch-1.12.1+cu116-cp38-cp38-win_amd64.whlpip install torchvision-0.13.1+cu116-cp38-cp38-win_amd64.whl

 

 cuda可以用,代码已经可以正常执行

来源地址:https://blog.csdn.net/zcs2632008/article/details/127025294

--结束END--

本文标题: CUDA安装教程

本文链接: https://www.lsjlt.com/news/390136.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
猜你喜欢
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作