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