Python——PyTorch的安装

                                     Python——PyTorch的安装

 

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

 

注意:

  • 查看cuda版本:
  • 方法1: cat /usr/local/cuda/version.txt
    方法2: nvcc --version

     

  • 查看cudnn版本:
  • cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

     

 

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

pip install torch torchvision torchaudio

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

 

验证一下:

 

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页