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

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐