3090下配dgl+torch环境

3090不能使用cuda11.0以下的版本,花了大半天终于成功搞定dgl和torch的环境配置。主要是遇到一些cuda问题,期间明白了:

  1. 一个机器可以装多个cuda,修改.bashrc的path可以调整当前使用的cuda。而不影响其他用户。
  2. /usr/local/cuda的cuda只是个软链接,cuda-10.2这种路径才是真正存的路径。
  3. nvidia-smi和nvcc-V显示的可能不一样,没什么影响,而且这两者可能和python进去 import torch print(torch.version.cuda)显示的cuda版本还不一样,想跑通代码,关注好print(torch.version.cuda)的版本即可。
  4. 用https://developer.nvidia.com/cuda-toolkit-archive下载的cudatoolkit可能还会缺libcusparse.so.10,可以通过conda install -c pytorch cudatoolkit=10.2在conda当前环境下载,然后按下面这个链接:https://blog.csdn.net/sunyueqinghit/article/details/115821287解决问题

配环境的顺序:

  1. pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple

  2. pip install -r requirement -i https://pypi.tuna.tsinghua.edu.cn/simple
    requirement:注意下dgl-cu111的写法
    scipy==1.5.2
    numpy==1.19.1
    networkx==2.5
    scikit-learn~=0.23.2
    matplotlib==3.4.1
    ogb==1.3.1
    dgl-cu111==0.6.1
    dgllife==0.2.6

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值