为了安装这个环境我真的是踩了不少坑,现在写下来记录一下。
就是CUDA的版本问题,在NVIDIA控制面板查找CUDA最大支持版本后不要立马安装那个版本,因为后续环境mxnet暂时没出CUDA11版本的,mxnet的版本和CUDA必须一一对应,导致你会安装不了mxnet。(pytorch同理)
mxnet用这个下:
pip install mxnet-cu102==1.7.0 -f https://dist.mxnet.io/python/cu102
为了安装这个环境我真的是踩了不少坑,现在写下来记录一下。
就是CUDA的版本问题,在NVIDIA控制面板查找CUDA最大支持版本后不要立马安装那个版本,因为后续环境mxnet暂时没出CUDA11版本的,mxnet的版本和CUDA必须一一对应,导致你会安装不了mxnet。(pytorch同理)
mxnet用这个下:
pip install mxnet-cu102==1.7.0 -f https://dist.mxnet.io/python/cu102