1.步骤一: 查看/更新电脑NVIC驱动版本:
(1)方法1
NVIC控制面板--系统信息--组件 有驱动(driver)版本(这里是12.0),驱动对cuda是向下兼容的,如果显示12.0,用我们常说的话就是最大支持12.0的cuda,也可以是11.8 11.6......的cuda。
(2)方法2:打开黑窗口
win+r --输入 cmd ,输入 nvidia-smi 查看cuda驱动版本(这里是12.0)
如果觉得驱动版本低,可以去NVIDIA官网下载最新版本
如:
2.步骤二:安装Anaconda
注意anaconda版本要和python版本对应(这里安装的最新版本的anaconda对应python3.9),如果没有python,安装anaconda时可以勾选对python的regester;如果有对应版本的python,建议下载与已有python版本对应的anaconda,此时在安装anaconda时可以不勾选对python的regester。
(查看已有python:win+r打开黑窗口--输入 python
查看python版本,这里是3.9.3)
anaconda官网--安装windows版本
3.步骤三:下载pytorch
打开anaconda下的黑窗口
输入
conda create --name pytorch python=3.9
创建虚拟空间,name是pytorch,python版本是3.9,下载需要的组件
输入
conda activate pytorch
激活虚拟空间:当(base)变为(pytorch)说明已激活
过程中遇到
pytorch官网--安装pytorch以前版本(之前安装最新的出了点问题)
注意安装的时候需要根据步骤一里查看的CUDA驱动版本,这里安装的版本不要高于步骤一的CUDA驱动版本(12.0)
下载该版本会自动安装CUDAtoolkit,复制该命令至黑窗口,程序下载
可以看到CUDAtoolkit、pytorch及其他组件将会下载
当遇到: Proceed([y]/n) ? y--yes,n--no
下载中,Pytorch还是比较大的,耐心等待一下,下载时间比较长
下载完成后检验安装: 如下即安装成功。
(该图参考(85条消息) 安装PyTorch详细过程_MC云鸷骚峰的博客-CSDN博客_pytorch安装)
希望对你有所帮助。