pytorch的安装及我出现的问题

强烈建议看土堆的教程

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所在的文件夹全部卸载,认认真真跟着土堆的教程再做一遍(大概半个小时)

你可能感兴趣的:(pytorch)