使用python进行并行计算编程时必须安装pycuda,在TX2使用命令“pip install pycuda”安装pycuda时一直安装失败,错误如下:

于是将pycuda下载下来进行安装,成功了!

pycuda下载链接:链接: https://pan.baidu.com/s/1F2yIYvE_iH6dFy_7hcAgQg 提取码: dipn 

下载完成后,解压文件,配置cuda的路径,命令如下:

python3 configure.py --cuda-root=/usr/local/cuda-10.2

使用以下命令进行安装:(安装前需安装setup工具)

sudo python3 setup.py install

整个安装过程时间有点,包括编译和安装。

setup工具包:

链接: https://pan.baidu.com/s/1fWirw4jSBmmSWKUcagdJpg 提取码: vwer 

安装方式:解压后,打开目录,终端输入命令:

sudo python setup.py install

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐