Ubuntu下安装cuda,cudnn和pytorch
Ubuntu下安装cuda,cudnn和pytorch服务器配置cuda和cudnn安装在安装一台电脑时候,遇到`'torch.cuda.is_available()`',输出是False的情况。
服务器配置
公司新配了几台2080TI主机打算做分布式训练,我事先以安装好Anaconda和pycharm了,这2个安装不难去官网下载对应安装就行了,下面主要记录cuda10.1、cudnn7.6和torch1.6的安装。至于为啥是这些,是因为要跑yolov5啊。。。
cuda和cudnn安装
新安装的Ubunt
原创
2020-08-26 17:26:13 ·
659 阅读 ·
1 评论