pytorch安装

先安装Anaconda，记得将

image-20240422165526309

添加到Path系统变量当中

然后安装CUDA和CuDNN

查看版本信息 为11.6

image-20240422165803294

对官网下载对应的版本https://developer.nvidia.com/cuda-toolkit-archive

这里下载11.6.0

image-20240422165924017

正常安装即可，要记住安装目录

image-20240422170657509

cmd输入nucc -V 验证是否安装成功

image-20240422170025901

然后安装CuDNN

下载地址：https://link.zhihu.com/?target=https%3A//developer.nvidia.com/rdp/cudnn-download

等待下载完成后进行解压，得到一个cuda文件夹，进入之后，全选，复制到之前CUDA所安装的文件夹下，有重复的进行替换即可。(这个步骤很乱，这两个都没安装指定的安装目录安装，我胡乱复制的)，最后通过图标打开文件位置找到的……

CUDA：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

结果验证

.\deviceQuery.exe

image-20240422171818603

.\deviceQuery.exe

image-20240422191344582

出现Result = PASS 代表成功

gpu版Pytorch安装

打开Anaconda Prompt(anaconda) 管理员模式打开

执行

conda env list

image-20240422191818378

创建一个新的虚拟环境
conda create -n new python=3.8.0

image-20240422191909363

此时查看，已创建

image-20240422191959014

激活并进入

conda activate new

查看当前拥有的包

conda list

image-20240422195316280

安装numpy

image-20240422195533951

pytorch下载：https://link.zhihu.com/?target=https%3A//pytorch.org/get-started/locally/

选择适合自己的版本

image-20240422195838613

复制安装命令到

image-20240422195906172

安装完成

image-20240422210352634

验证

image-20240422210443523

posted @ 2024-04-24 16:31  清澈的澈  阅读(8)  评论(0)  编辑  收藏  举报