cuda pytorch安装详细教程 GPU版

1.先安装anaconda Anaconda
1.1这里用了最简单的方法，后面将pytorch直接安装到base配置下面了。

1.2下载安装，全部勾选。
一定添加文件夹到path中去，否则后面下载pytorch，无法传输。

1.3 通过cmd，输入conda --version ， 测试结果。

2.安装nvidia驱动，cuda 11.6
cuda安装

2.1添加文件夹到系统path中
2.3 通过cmd，nvcc -V 查看cuda版本是否正确。

这里装的是cuda_11.6

3.安装pytorch

地址 ：pytorch.org

在这里选择自己需要匹配的版本

// CUDA 11.6
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

4. 测试

输入ipython和三行测试代码，为True就是正确了。

import torch

print(torch.cuda.is_available())
print(torch.__version__)