服务器安装cuda版本的pytorch+DGL

1、先创建pytorch环境:conda create -n ljj_torch112 python=3.8

看本机的:

服务器安装cuda版本的pytorch+DGL_第1张图片

先看自己的cuda版本:(最权威的看:nvcc --version)

服务器安装cuda版本的pytorch+DGL_第2张图片

 10.0的cuda于是不太符合,所以换一个10.2的cuda比较常用!

创建pytorch环境:

(资料:(258条消息) Ubuntu中安装Pytorch_ubuntu安装pytorch_不写八个的博客-CSDN博客)

服务器安装cuda版本的pytorch+DGL_第3张图片

 服务器安装cuda版本的pytorch+DGL_第4张图片

 2、下载pytorch(带上了cuda10.2)

我要下载的是1.12的,不是2.0的新版,所以到这里去找以前的pytorch:

Start Locally | PyTorch

conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c pytorch

服务器安装cuda版本的pytorch+DGL_第5张图片服务器安装cuda版本的pytorch+DGL_第6张图片

服务器安装cuda版本的pytorch+DGL_第7张图片

 2.1 查看torch、cuda等的版本

(258条消息) 在Python中,查看torch、cuda和cudnn的版本_cudnn版本查看_初识-CV的博客-CSDN博客

服务器安装cuda版本的pytorch+DGL_第8张图片

 3、下载DGL

Deep Graph Library (dgl.ai)

服务器安装cuda版本的pytorch+DGL_第9张图片

 conda install -c dglteam/label/cu102 dgl

服务器安装cuda版本的pytorch+DGL_第10张图片

 3.1  可以用conda list查看DGL是否下载成功

服务器安装cuda版本的pytorch+DGL_第11张图片

 服务器安装cuda版本的pytorch+DGL_第12张图片

 

你可能感兴趣的:(深度学习,人工智能)