本文介绍在anaconda中安装pytorch。
最近因为学习需要,要下载pytorchGPU版本来训练网络。相信pytorch大家都不陌生了,PyTorch 是一个 Torch7 团队开源的 Python 优先的深度学习框架,提供两个高级功能:
- 强大的 GPU 加速 Tensor 计算(类似 numpy)
- 构建基于 tape 的自动升级系统上的深度神经网络
于是我在网上查了对应版本,在anaconda中创建了虚拟环境,然后到pytorch官网复制了conda命令安装,结果却遇到许多问题。上网查了查,很多文章都是修改源,或者更换镜像,然后增加报错时间等方法,然而许多我尝试之后都不成功,出现各种报错,还浪费很多时间。最后直接下载对应的whl文件,再本地安装,最后也是成功安装上了。
第一步:查看电脑对应显卡cuda版本
win+r打开运行,输入cmd打开命令行,输入nvidia-smi,查看cuda版本。(前提是电脑里有GPU)
我的cuda版本是11.7
第二步:在下面网站中找到对应的torch和torchvision,下载whl文件
点击torch和