python3.7,显卡CUDA版本11.1,安装pytorch-CUDA

Pytorch安装真的折磨人,cpu版本的安装完,跑起来贼慢,要安装GPU版本的。显卡CUDA版本比较老了11.1的。

通过在CMD 输入nvidia-smi命令可以在第一行最右边看到CUDA版本。

python3.7,显卡CUDA版本11.1,安装pytorch-CUDA_第1张图片

 现在Pytorch最新版本不支持CUDA11.1一下的版本了,最低支持到11.6

python3.7,显卡CUDA版本11.1,安装pytorch-CUDA_第2张图片

现在早期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

多的就不说了,希望这个命令能长时间起作用,如果版本更新,原先的版本又受影响,还得找新的命令。

你可能感兴趣的:(pytorch,深度学习,python)