目录
前言
本文针对的是有英伟达的显卡进行GPU的pytorch安装。综合许多安装pytorch的方法感觉用conda指令安装较为简单,所以本文采用的是conda安装。
如果最后返回的is_available()返回的是False,大家可以尝试以下步骤进行排除:
1、首先确定显卡是否支持CUDA(https://www.geforce.com/hardware/technology/cuda/supported-gpus),或者看本文后面介绍的方法验证
2、确保驱动升级到最新(可以看我最后的解决方法)
一、Anaconda安装
anaconda的安装参考:windows安装anaconda最详细教程_light-124的博客-CSDN博客_windows安装anaconda
二、CUDA
找到NVIDIA的控制面板,查看自己的驱动版本
查看CUDA的版本,这里看到是11.7
或者win+r输入cmd中输入
nvidia-smi
三、安装pytorch
1.创建虚拟环境
打开anaconda prompt,输入conda create -n pytorch python=3.9(我是命名的pytoch,