一、conda指令创建pytorch环境
首先需要下载Anaconda
以下命令在 Anaconda Prompt 中输入:
conda create -n DTIAM python=3.10
二、下载Pytorch离线安装包
我们先去官网pytorch查找你电脑对应的GPU版本(cuda版本),torch版本,torchvision版本,torchaudio版本,python版本。注意:这几个版本一定要一一对应!!!
如果网页进不去就翻墙
首先查看自己电脑CUDA版本:
打开自己电脑的CMD,然后输入
nvidia-smi
CUDA Version是你电脑显卡支持的最高cuda版本,我们只要下载的cuda版本比它稍微低点或者等于它就行,注意不要太低,太低会不支持了。
然后我们去官网下载地址去下载对应的torch,torchvision,torchaudio。
下载地址可以是前面网址查到的torch,torchvision,torchaudio对应关系后面的下载网址,或者点击下面的下载网址都可以。
进入下图这三个链接下载上图中对应的版本
torch=2.5.0 torchvision=0.20.0 torchaudio=2.5.0
尽量将这三个文件下载至一个地方
注意:cu后面的表示GPU版本,cp后面的表示python版本,win表示wins系统
如果需要其他cpu或者其他系统版本的在里面都可以自己找到
三、安装
然后第一步我们打开conda,激活我们开头创建的环境
conda activate DTIAM
第二步进入到那三个文件下载的目录,假如我的文件在F盘,先进入F,再进入三个文件所在的目录
第三步开始依次安装torch=2.5.0 torchvision=0.20.0 torchaudio=2.5.0(前提是要有网!不需要断网!因为安装过程中还会自动安装一些其他文件):
pip install torch-2.5.0+cu118-cp310-cp310-win_amd64.whl
pip install torchvision-0.20.0+cpu-cp310-cp310-win_amd64.whl
pip install torchaudio-2.5.0+cpu-cp310-cp310-win_amd64.whl
最后恭喜你安装完成!
DGL安装:dgl与cuda对应的版本参考如下网站:https://www.dgl.ai/pages/start.html
然后输入对应的命令即可