Windows下conda安装pytorch GPU版

1.安装miniconda,不细讲了,自己去百度,miniconda自带python,可以通过conda创建虚拟python环境,安装Pytorch的话建议python版本大于3.8,完成后注意配置环境变量。

2.安装CUDA:

查看自己CUDA版本,Nvidia控制面板中找,不再赘述。根据查看的版本,下载 CUDA Toolkit并安装。安装完成记得配置系统环境变量,这里自己百度。如何验证安装成功呢?

终端测一下:

方法一:

nvcc -V

Windows下conda安装pytorch GPU版_第1张图片

方法二:

nvidia-smi

Windows下conda安装pytorch GPU版_第2张图片

3.GPU加速,下载

你可能感兴趣的:(深度学习,conda,pytorch,人工智能)