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
- 测试
输入ipython和三行测试代码,为True就是正确了。
import torch
print(torch.cuda.is_available())
print(torch.__version__)