mxnet以及dgl的配套安装(linux服务器、GPU版)
1、在确定CUDA正常能够使用的情况下,先查看CUDA的版本,linux服务器上使用 nvcc --version 命令查看,我的是10.1版本的。 2、进入mxnet官网: https://mxnet.apache.org/versions/1.9.0/get_started?platform=linux&language=python&processor=gpu&environ=pip& 根据自身版本的CUDA以及python版本安装GPU版的mxnet,我的python版本是3.6的,对照表格我可以选择安装1.7.0版本,使用命令 pip install mxnet-cu101==1.7.0 进行安装。 ![在这里插入图片描述](https://img-blog.csdnimg.cn/6da852e241774e3a876bea59ac2ce617.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBAd2VpeGluXzQ0MjMyNjM5,size_19,color_FFFFFF,t_70,g_se,x_16 3、安装GPU版本的dgl。进入官网 https://www.dgl.ai/pages/start.html ,根据自身的CUDA版本以及python进行选择安装(系统会自动匹配安装相应版本的dgl)
4、若要安装指定GPU版本的dgl,可以使用 在上述命令中指定版本,例如 :conda install -c dglteam dgl-cuda10.2==0.4.3post2 ,如果使用该种方法不成功。可以选择以下方法: 我的CUDA版本是10.1 ,进入网址 https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu101/ CUDA版本是10.2 ,进入网址 https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu102/
此时我需要在linux 64系统上安装的是 0.4.3post1 版本,此时python版本为3.6。找到所需版本的dgl后下载。然后在文件所在的文件路径在使用命令 pip install dgl_cu101-0.4.3.post1-cp36-cp36m-manylinux1_x86_64.whl 进行安装。
|