pytorch-v2.0.1 cuda arm64 aarch64 torch 2.0.1+cu118 源码编译笔记【2】验证cuda安装 成功

接上篇

pytorch-v2.0.1 cuda arm64 aarch64 torch 2.0.1+cu118 源码编译笔记_hkNaruto的博客-CSDN博客

由于采用/usr/local/bin/gcc编译,先设置LD_LIBRARY_PATH,再启动python3

复制代码
export LD_LIBRARY_PATH=/usr/local/lib64:/usr/local/lib:/usr/lib64:/usr/lib
/usr/local/Python-3.10.12/bin/python3

import torch报错

/usr/local/Python-3.10.12/bin/python3

Python 3.10.12 (main, Sep 4 2023, 10:01:29) [GCC 9.2.0] on linux

Type "help", "copyright", "credits" or "license" for more information.

>>> import torch

No sympy found

Traceback (most recent call last):

File "<stdin>", line 1, in <module>

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/init.py", line 1465, in <module>

from . import _meta_registrations

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_meta_registrations.py", line 7, in <module>

from torch._decomp import _add_op_to_registry, global_decomposition_table, meta_table

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_decomp/init.py", line 169, in <module>

import torch._decomp.decompositions

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_decomp/decompositions.py", line 10, in <module>

import torch._prims as prims

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_prims/init.py", line 33, in <module>

from torch._subclasses.fake_tensor import FakeTensor, FakeTensorMode

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_subclasses/init.py", line 3, in <module>

from torch._subclasses.fake_tensor import (

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_subclasses/fake_tensor.py", line 13, in <module>

from torch._guards import Source

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_guards.py", line 78, in <module>

class ShapeGuard(NamedTuple):

File "/usr/local/Python-3.10.12/lib/python3.10/site-packages/torch/_guards.py", line 79, in ShapeGuard

expr: sympy.Expr
NameError: name 'sympy' is not defined

安装sympy

在线

复制代码
/usr/local/Python-3.10.12/bin/python3 -m pip install sympy

离线

复制代码
[root@ceph3 pytorch-libs]# /usr/local/Python-3.10.12/bin/python3 -m pip install *.whl
Processing ./filelock-3.12.3-py3-none-any.whl
Processing ./Jinja2-3.1.2-py3-none-any.whl
Processing ./MarkupSafe-2.1.3-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl
Processing ./mpmath-1.3.0-py3-none-any.whl
Processing ./networkx-3.1-py3-none-any.whl
Processing ./sympy-1.12-py3-none-any.whl
Requirement already satisfied: typing-extensions>=4.7.1 in /usr/local/Python-3.10.12/lib/python3.10/site-packages (from filelock==3.12.3) (4.7.1)
mpmath is already installed with the same version as the provided wheel. Use --force-reinstall to force an installation of the wheel.
sympy is already installed with the same version as the provided wheel. Use --force-reinstall to force an installation of the wheel.
Installing collected packages: networkx, MarkupSafe, filelock, Jinja2
Successfully installed Jinja2-3.1.2 MarkupSafe-2.1.3 filelock-3.12.3 networkx-3.1
WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager. It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv

[notice] A new release of pip is available: 23.0.1 -> 23.2.1
[notice] To update, run: /usr/local/Python-3.10.12/bin/python3 -m pip install --upgrade pip

import torch成功,测试cuda是否可用

参考:

PyTorch快速安装并验证GPU是否可用_pytorch测试gpu_ai_patch的博客-CSDN博客

相关推荐
葫三生22 分钟前
如何评价《论三生原理》在科技界的地位?
人工智能·算法·机器学习·数学建模·量子计算
智者知已应修善业1 小时前
【51单片机用数码管显示流水灯的种类是按钮控制数码管加一和流水灯】2022-6-14
c语言·经验分享·笔记·单片机·嵌入式硬件·51单片机
m0_751336391 小时前
突破性进展:超短等离子体脉冲实现单电子量子干涉,为飞行量子比特奠定基础
人工智能·深度学习·量子计算·材料科学·光子器件·光子学·无线电电子
美狐美颜sdk4 小时前
跨平台直播美颜SDK集成实录:Android/iOS如何适配贴纸功能
android·人工智能·ios·架构·音视频·美颜sdk·第三方美颜sdk
孞㐑¥5 小时前
Linux之Socket 编程 UDP
linux·服务器·c++·经验分享·笔记·网络协议·udp
DeepSeek-大模型系统教程5 小时前
推荐 7 个本周 yyds 的 GitHub 项目。
人工智能·ai·语言模型·大模型·github·ai大模型·大模型学习
郭庆汝5 小时前
pytorch、torchvision与python版本对应关系
人工智能·pytorch·python
小雷FansUnion7 小时前
深入理解MCP架构:智能服务编排、上下文管理与动态路由实战
人工智能·架构·大模型·mcp
资讯分享周7 小时前
扣子空间PPT生产力升级:AI智能生成与多模态创作新时代
人工智能·powerpoint
sealaugh327 小时前
aws(学习笔记第四十八课) appsync-graphql-dynamodb
笔记·学习·aws