0
点赞
收藏
分享

微信扫一扫

深度学习的环境minitorch3+CUDA10.2+cudnn+mxnet1.7的坑

凉夜lrs 2022-04-03 阅读 88

为了安装这个环境我真的是踩了不少坑,现在写下来记录一下。

就是CUDA的版本问题,在NVIDIA控制面板查找CUDA最大支持版本后不要立马安装那个版本,因为后续环境mxnet暂时没出CUDA11版本的,mxnet的版本和CUDA必须一一对应,导致你会安装不了mxnet。(pytorch同理)

mxnet用这个下:

pip install mxnet-cu102==1.7.0 -f https://dist.mxnet.io/python/cu102

举报

相关推荐

0 条评论