ubuntu18.04系统3060显卡配置3D点云深度学习环境CUDA11.3+Cudnn8.2.1+anaconda3.5.2+pytorch+apex1.0+spconv1.2.1

最新推荐文章于 2024-03-23 09:59:55 发布

莫识丶丶

最新推荐文章于 2024-03-23 09:59:55 发布

阅读量2k

点赞数 2

文章标签： pytorch 深度学习 tensorflow 人工智能

本文链接：https://blog.csdn.net/weixin_49854786/article/details/122340939

版权

ubuntu18.04系统3060显卡配置3D点云深度学习环境CUDA11.3+Cudnn8.2.1+anaconda3.5.2+pytorch+apex1.0+spconv1.2.1@TOC

apex与spconv一定要安装以下版本，否则会有很多莫名其妙的错误，本人小白一个，历时一个月终于完成这个环境配置
参考博主连接如下：感谢大佬！！！

安装显卡驱动+CUDA11.3+Cudnn8.2.1

参考博客安装显卡驱动
选择对应版本，我安装完成后显示
在这里插入图片描述

anaconda3.5.2

去清华镜像源下载
1.到下载目录下

bash Anaconda3-5.2.0-Linux-x86_64.sh

2.最后询问是否install VS code？输入no

pytorch 1.10.0

注意：不要使用conda安装，会出现solve environment ：failed
先进入环境：

conda activate name 
#name 是创建的环境名

在创建的环境下输入：

pip3 --default-timeout=60000 install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple --trusted-host pypi.tuna.tsinghua.edu.cn

conda所创建的cudatoolkit环境版本一定要与电脑base环境下的CUDA版本一致，否则后期会出错
安装完成后测试pytorch

python

在这里插入图片描述

apex1.0

依照github官网安装

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

如果torch版本不匹配的话第三步会报错
测试是否安装成功
在这里插入图片描述

spconv1.2.1

spconv1.2.1地址
下载之后

cd spconv1.2.1
cd third_party
git clone https://github.com/pybind/pybind11
cd ..
python setup.py bdist_wheel
cd ./dist   
pip install spconv-1.2.1-cp36-cp36m-linux_x86_64.whl

最后会显示success install spconv1.2.1