一、配置TensorRT
- 解压安装文件:
| cd ~/0-software/ |
| tar xzvf TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-11.8.tar.gz |
- 运行
sudo gedit ~/.bashrc
,在文档中添加:
| export PATH=/home/wangbo/0-software/TensorRT-8.6.1.6/bin:$PATH |
| export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/wangbo/0-software/TensorRT-8.6.1.6/lib |
二、配置PyTorch
- 新建conda环境
| conda create -n base_113 |
| conda activate base_113 |
| |
- 安装PyTorch 1.13.1
| conda install pytorch==1.13.1 torchvision==0.14.1 pytorch-cuda=11.6 -c pytorch -c nvidia |
- 安装pycuda
去whl下载网站,根据cuda和python确定版本号,例如:2022.1
然后运行安装命令:
| pip install cuda-python |
| pip install pycuda == 2022.1 |
- 安装TensorRT
| pip install tensorrt==8.6.1 |
【推荐】中国电信天翼云云端翼购节,2核2G云服务器一口价38元/年
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 在.NET Core中使用异步多线程高效率的处理大量数据
· 聊一聊 C#前台线程 如何阻塞程序退出
· 几种数据库优化技巧
· 聊一聊坑人的 C# MySql.Data SDK
· 使用 .NET Core 实现一个自定义日志记录器
· 字节豆包,来园广告
· 我用cursor, 半就开发了一个手机壁纸小程序,真的太强了
· 聊一聊 C#前台线程 如何阻塞程序退出
· 订单超时自动取消,我们是这样做的。。。
· C#字符串拼接的几种方式及其性能分析对比