IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> 利用torch安装CuPy,使得numpy(np)在CUDA(gpu)上加速。 -> 正文阅读

[人工智能]利用torch安装CuPy,使得numpy(np)在CUDA(gpu)上加速。

前言

标题名字是很奇怪,没关系,重点看内容。

正常安装CupPy需要较为复杂的cuda环境配置,可以参考文章——CUDA环境配置

如果你觉得去官网安装CUDA Toolkit太麻烦,那我们可以利用pyotch的cudatookit来保证Cupy的正常运行。

正文

  1. CuPy官网
    官网给出了详细的安装操作,但是需要手动安装CUDA Toolkit,如果你没有实践过,这也许会比较难。
    官网给出了相应版本对应的安装命令:
    在这里插入图片描述
    我电脑是11.1,所以这里执行下面命令即可
pip install cupy-cuda111

安装好以后直接去运行肯定是不行的,会报错:

Failed to import CuPy.

If you installed CuPy via wheels (cupy-cudaXXX or cupy-rocm-X-X), make sure that the package matches with the version of CUDA or ROCm installed.

On Linux, you may need to set LD_LIBRARY_PATH environment variable depending on how you installed CUDA/ROCm.
On Windows, try setting CUDA_PATH environment variable.

Check the Installation Guide for details:
  https://docs.cupy.dev/en/latest/install.html

Original error:
  ImportError: DLL load failed while importing runtime: 找不到指定的模块。

意思很明确了,就是说你没有安装cuda,或者没有指定CUDA_PATH的环境变量。

  1. 如果你安装了pytorch的cudatoolkit, 你会发现你是可以使用cuda加速的,那为什么cupy就不可以呢?
    因为在pytorch官网安装的cudatoolkit并不是官网安装CUDA Toolkit,它们是不相同的。
    pytorch官网安装的cudatoolkit是一些编译好的CUDA程序,当系统上存在兼容的驱动时,这些程序就可以直接运行。安装pytorch会同时安装cudatoolkit,且pytorch的GPU运算直接依赖cudatoolkit,因此无需安装CUDA Toolkit即可使用pytorch进行深度学习。

  1. 对于我们这种喜欢偷懒的人,肯定是选择安装pytorch的cudatoolkit啊,毕竟一行命令就都安装好了。
    对于pytorch和cuda的安装,可以参考我的另外一篇文章——pytorch+cuda,各种较老版本cuda,两步完成

  2. 对于安装好了pytorch的cudatoolkit以后,怎么让CuPy在cudatoolkit上运行呢,这就很容易了,只需要先导入一下torch就可以了!

放个下图给你们看看,虽然还是会Warning,但是已经可以正常使用了。
在这里插入图片描述

总结一下

  1. 首先安装pytorch和cudatoolkit,可以参考pytorch+cuda,各种较老版本cuda,两步完成
  2. 安装cupy,通过上面文章也可以找到cuda的版本,然后pip install cupy-cuda??? (???就是版本)
  3. 记得在import cupy 之前先 import torch就可以了。
import torch
import cupy

然后就可以使用cupy了。

  1. 可以使用
a = cupy.array([1,2])
print(a.device)

验证一下是否导入成功,如果成功了就成功了,如果不成功,去官网安装CUDA Toolkit吧。

  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2022-05-13 11:44:22  更:2022-05-13 11:47:32 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/26 5:26:57-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码