强烈建议看土堆的教程
https://www.bilibili.com/video/BV1S5411X7FY?p=28&vd_source=093939ab85e6e431b372821203a2d7a6
创建虚拟环境
显卡驱动版本大于cuda runtime
显卡驱动可以让电脑识别显卡,cude runtime可以让软件调用显卡
显卡算力8.6
确定cuda runtime版本
确定cuda driver version 11.5
最终确定pytorch版本小于等于11.5
要知道的一些
清华镜像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
(Ctrl加f,edge内查找)
我的网速慢所以用镜像,推荐用官网(如果网速好)
验证pytorch
先进入虚拟环境,再输入conda list看看有没有pytorch或torch
再输入python
再输入import torch
再输入torch.cuda.is_available()
出现Ture说明安装成功
关于我这个小白出现的一些问题
1.我第一次安pytorch时网不好中间卡了最后验证的时候出错了
2.我下载的出现红码报错了
我的解决方法(一个纯小白的方法,欢迎随时补充)
把anaconda所在的文件夹全部卸载,认认真真跟着土堆的教程再做一遍(大概半个小时)