悉知网站建设郑州网站建设哪里好

1. 安装anaconda

地址 ：Anaconda | The World's Most Popular Data Science Platform

安装选项全默认点击next就行。

2. 查看支持cuda版本

cmd命令行输入nvidia-smi。下图右上角显示11.6为支持的cuda版本。

要是显示没有nvidia-smi命令。得安装nvidia驱动，一般情况都有。

3. 创建conda环境

在windows搜索栏找到anaconda prompt。打开它。

4. 创建新conda环境

conda create -n torch_env(这个是环境名) python==3.8(指定python版本，看个人需求)

5. 进入conda环境

conda activate torch_env(环境名)

6. 安装cudatoolkit

conda search cudatoolkit （会显示一堆版本，填与cuda版本一致的。）

conda install cudatoolkit=11.6.0（注意，填cuda版本一致的）

7. 安装cudnn

conda search cudnn

conda install cudnn=8.4.1.50(选最新的)

8. 安装pytorch

上官网 PyTorch

这么选。复制代码，运行。

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu116

9. 验证

安完了验证一下。这个不用多说了把。

python -c "import torch; print(torch.cuda.is_available())"

python -c "import torch; print(torch.__version__)"

python -c "import torch; print(torch.version.cuda)"

附加：

1. linux安装类似，上述代码打在shell终端上就行。

区别在于linux得有nvidia显卡驱动的先，再输入nvidia-smi查看cuda。

2. 关闭conda环境

conda deactivate

3. 安装pytorch-geometric （个人用途）

Installation — pytorch_geometric documentation (pytorch-geometric.readthedocs.io)

pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.13.0+cu116.html

安完检验

import torch
from torch_geometric.data import Dataedge_index = torch.tensor([[0, 1, 1, 2],[1, 0, 2, 1]], dtype=torch.long)
x = torch.tensor([[-1], [0], [1]], dtype=torch.float)data = Data(x=x, edge_index=edge_index)
print(data)