Python——PyTorch的安装_error: torch 2.0.0+cu118 is installed but torch==2

作者：盐析白兔 | 2024-07-13 04:35:08

踩

error: torch 2.0.0+cu118 is installed but torch==2.0.1 is required by {'torc

越来越多的人选择PyTorch的深度学习框架了，那么这么包如何安装呢？注意，它不能像Tensorflow或者Keras那样安装了，否者会报错的。官网给了具体说明，咱们移步：PyTorch官网

注意：


方法1: cat /usr/local/cuda/version.txt
方法2: nvcc --version

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

比如，我的环境是Mac，无CUDA，pip安装，那就直接是：

pip install torch torchvision torchaudio

如果安装比较慢，配置一下镜像源，参考以前的笔记：Mac——三步更改Python 默认pip源

验证一下：

声明：本文内容由网友自发贡献，转载请注明出处：【wpsshop】