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

凉夜lrs

关注

阅读 88

2022-04-03

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

就是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)

0 0 举报