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

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

参考博客安装显卡驱动
选择对应版本,我安装完成后显示
ubuntu18.04系统3060显卡配置3D点云深度学习环境CUDA11.3+Cudnn8.2.1+anaconda3.5.2+pytorch+apex1.0+spconv1.2.1_第1张图片ubuntu18.04系统3060显卡配置3D点云深度学习环境CUDA11.3+Cudnn8.2.1+anaconda3.5.2+pytorch+apex1.0+spconv1.2.1_第2张图片

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

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

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版本不匹配的话第三步会报错
测试是否安装成功
ubuntu18.04系统3060显卡配置3D点云深度学习环境CUDA11.3+Cudnn8.2.1+anaconda3.5.2+pytorch+apex1.0+spconv1.2.1_第4张图片

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

你可能感兴趣的:(pytorch,深度学习,tensorflow,人工智能)