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

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

更多推荐