Tensor安装和测试

1: 打开git官方

https://github.com/NVIDIA/TensorRT

2: 下载得到:TensorRT-10.2.0.19.Linux.x86_64-gnu.cuda-11.8.tar.gz

3: 下载后配置环境变量,上面地址记得改成真实地址。

4: 如果想python使用tensorrt,那么 解压后目录,再进入到python目录

5: 看到很多版本,查看当前python版本

6: 所以运行。(cp311,是指python3.11)

pip3 install tensorrt-10.2.0-cp311-none-linux_x86_64.whl

验证tensor版本:

安装完成后验证:

1:走进sample.

2: 执行:

make clean

make

3: 然后回去bin目录

成功:

相关推荐
RobinDevNotes11 小时前
模型训练工程师必须搞懂的DDP和FSDP
人工智能·pytorch
洛阳纸贵11 小时前
AI-PyTorch(一)基础代码实操和自动求导
人工智能·pytorch·python
磁场转动100万匹12 小时前
PyTorch 食物图像分类实战:从数据准备到 CNN 训练全流程解析
pytorch·分类·cnn
ai小陈13 小时前
PyTorch梯度累积与裁剪实战:小显存也能稳定训练大批次
人工智能·pytorch·python·深度学习·ai·gpu算力
C^h1 天前
pytorch 适合初学者 0基础学习
人工智能·pytorch·python
论文复现现场1 天前
本地 PyTorch 训练 OOM,第一次租 RTX 4090 云 GPU 怎么迁移项目?从环境检查到 100 Step 跑通
人工智能·pytorch·python·深度学习·cuda
2601_962099081 天前
在Python中使用LSTM和PyTorch进行时间序列预测
pytorch·机器学习·lstm·时间序列预测·数据预处理
论文复现现场1 天前
单卡RTX 3090能训练,切到4卡却OOM:Accelerate多卡训练怎么排查?
人工智能·pytorch·云计算·gpu算力·多卡训练
小白说大模型1 天前
去AI味提示词大全:25个实用Prompt帮你降低AI率
大数据·人工智能·pytorch·深度学习·机器学习·prompt
Hali_Botebie1 天前
PyTorch 内存布局,.view()要合并哪两个维度(比如 B 和 G),这两个维度在内存里就必须“紧挨着”。
人工智能·pytorch·python