建议使用CUDA10.2,pytorch1.7的组合
建议使用CUDA10.2,pytorch1.7的组合
建议使用CUDA10.2,pytorch1.7的组合
本机配置
外星人台式机R7、gtx1080ti
最终配置:
1、显卡驱动:460.56
2、cuda: release 10.2, V10.2.89,下载地址
3、cudnn:下载地址
./libcudnn8-dev_8.0.5.39-1+cuda10.2_amd64.deb
./libcudnn8-samples_8.0.5.39-1+cuda10.2_amd64.deb
./libcudnn8_8.0.5.39-1+cuda10.2_amd64.deb
4、pytroch:1.7.0 下载地址
# CUDA 10.2
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
5、其他
Package Version
----------------- -------------------
certifi 2021.5.30
dataclasses 0.8
future 0.18.2
MinkowskiEngine 0.5.4
mkl-fft 1.3.0
mkl-random 1.2.1
mkl-service 2.3.0
numpy 1.20.2
pip 21.1.3
setuptools 52.0.0.post20210125
six 1.16.0
torch 1.7.0
typing-extensions 3.10.0.0
wheel 0.36.2
一、下载相关nvidia显卡驱动、cuda、cudnn
cuda: release 10.2, V10.2.89,下载地址
cudnn:下载地址
./libcudnn8-dev_8.0.5.39-1+cuda10.2_amd64.deb
./libcudnn8-samples_8.0.5.39-1+cuda10.2_amd64.deb
./libcudnn8_8.0.5.39-1+cuda10.2_amd64.deb
二、安装驱动
推荐先安装NVIDIA显卡驱动
推荐先安装NVIDIA显卡驱动
推荐先安装NVIDIA显卡驱动
参考:
1、https://www.cnblogs.com/walker-lin/p/11142673.html
2、https://zhuanlan.zhihu.com/p/366882419
3、https://suyunzzz.github.io/2020/04/21/ubuntu%E5%AE%89%E8%A3%85/
三、安装pytorch、Minkowski Engine
# CUDA 10.2
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=10.2 -c pytorch
本地安装
参考:
MinkowskiEngine
MinkowskiEngine 0.5.4
# install system requirements
sudo apt install build-essential python3-dev libopenblas-dev
# Or if you want local MinkowskiEngine
git clone https://github.com/NVIDIA/MinkowskiEngine.git
cd MinkowskiEngine
export CXX=g++-7
python setup.py install
测试
参考:博客1