解决3080显卡tensorflow与pytorch在cuda11.1的安装问题
一、驱动及cuda、cudnn的安装
RTX3080采用了新的Ampere架构GA102-200,那么显卡驱动也必然是需要最新的,可以参考cuda与驱动对应的关系:
我安装的是455.23.04版本,CUDA是11.1版本,cudnn是8.0.4.30版本,具体的安装方法可参考一下网址。
ubuntu16.04系统run方式安装nvidia显卡驱动Ubuntu16.04下cuda和cudnn的卸载和升级
方法都是大同小异,最后输入nvidia-smi确认驱动是否安装好,nvcc --versio
转载
2020-11-16 10:38:51 ·
8113 阅读 ·
2 评论