Point_transformer环境搭建以及解决:No module named pointopscuda
ubuntu22,python3.8,cuda12.1,nvidia driver535,GeForce GTX 1080Ti
一、软件环境安装
1.安装ubuntu22
2.安装anaconda
3.安装python
二、虚拟环境搭建
1.创建python=3.8的虚拟环境
conda create -n trans python=3.8
2.打开虚拟环境
conda activate trans
3.安装cuda12.1.0版本
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda_12.1.0_530.30.02_linux.run
(这是下载cuda12.1安装包,推荐在bash环境中下载)
sudo sh cuda_12.1.0_530.30.02_linux.run
(这是在虚拟环境中安装cuda命令)
4.安装相对应pytorch,cuda 12.1版本
conda install pytorch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0 pytorch-cuda=12.1 -c pytorch -c nvidia
5.打开“软件和更新”,选择“附加驱动”为:nvidia-driver-535(专有,tested)
6.打开python调试代码即可
注:会提示找不到gcc与g++,这里按照提示安装默认gcc和g++即可,要求gcc和g++版本匹配,均为11.4.0
三:No module named pointopscuda
1.虚拟环境中运行打开目录代码:
cd lib/pointops
2.运行setup.py
python3 setup.py install
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)