赞
踩
Pytorch安装真的折磨人,cpu版本的安装完,跑起来贼慢,要安装GPU版本的。显卡CUDA版本比较老了11.1的。
通过在CMD 输入nvidia-smi命令可以在第一行最右边看到CUDA版本。
现在Pytorch最新版本不支持CUDA11.1一下的版本了,最低支持到11.6
现在早期pytorch版本安装有很多问题,在官网上的早期版本安装命令,现在大多数都有一些问题。使用pip安装命令,一直报错没有匹配的版本
无奈只能尝试用conda安装,但是虽然官网上的conda安装命令可以安装(conda环境不能换源,不然各种问题)但是其中安装的并不能使用CUDA,因为他是通过cudatoolkit来实现的,但是这个不知道为什么根本不起作用,最后还是改用pip安装命令。
经过红字提示,找到一个可以正常安装的版本命令1.10.1
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
多的就不说了,希望这个命令能长时间起作用,如果版本更新,原先的版本又受影响,还得找新的命令。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。