1.安装miniconda,不细讲了,自己去百度,miniconda自带python,可以通过conda创建虚拟python环境,安装Pytorch的话建议python版本大于3.8,完成后注意配置环境变量。
2.安装CUDA:
查看自己CUDA版本,Nvidia控制面板中找,不再赘述。根据查看的版本,下载 CUDA Toolkit并安装。安装完成记得配置系统环境变量,这里自己百度。如何验证安装成功呢?
终端测一下:
方法一:
nvcc -V
方法二:
nvidia-smi
文章来源:https://uudwc.com/A/ZGkeA
3.GPU加速,下载文章来源地址https://uudwc.com/A/ZGkeA