Ubuntu 20.04 + nvidia驱动11.4 + cuda11.0 + cudnn8.0.5 +pytorch 1.7.0(简易记录)
1. Ubuntu20.04
启动盘已拷贝好,以后可以直接用,就不再记录。
2.驱动470
会遇见驱动问题,默认test驱动版本,会有一个密码,记住,然后重启进入蓝色界面
3.cuda11.0
官网下载安装,runfile[load]->continue,安装时取消驱动
检查: XX_version.h | grep XX
Ubuntu安装cuda与cudnn,亲测可用_JulyLi2019的博客-CSDN博客_ubuntu安装cuda和cudnn
4.cudnn8.0.5
官网下载4个,Linux->cp,其他三个->sudo dpkg -i XX.deb
5.pytorch1.7.0
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0
-f https://download.pytorch.org/whl/torch_stable.html
最后,直接下载nnUNet,大部分医学分割要用到包都在说明中,无需自己再安装很多包。
(半年前安装时随手记下的,整理一下,不够细节,之后如果再安装再更新,以及如何给Ubuntu换衣服)