ubuntu18.04安装CUDA10.0
我为什么要安装CUDA呢?
因为跑很多深度学习项目的时候都会有device=“cuda:0”
CUDA(Compute Unified Device Architecture)是一种用于GPU并行计算框架,用了这个架构就可以把深度学习东西移到GPU上去跑,跑的更快。只有NVIDIA的显卡才能用这个框架,AMD的是用不了的。
为什么要安装CUDA10.0?
在做项目的时候我希望有一个py27的环境,因为只有python2才能兼容好ROS系统,试过python3可以发布话题等,但是在导某些包的时候还是会有问题,比如cv_bridge这个模块,有时候它会去找自带python2.7里面的环境。会比较乱。同时我又希望安装pytorch的gpu版本。
https://download.pytorch.org/whl/torch_stable.html
这里可以看支持python2.7的pytorch的版本
可以看到这里cuda10.0可以支持torch1.4最高就是1.4啦,所以我只能这样安装1