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
  1. 测试

输入ipython和三行测试代码,为True就是正确了。
在这里插入图片描述

import torch

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

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐