略
-
前言
- 就此项目而言,可自主选择安装与否
- 不安装可直接看第三部分
-
Anaconda的好处
- Anaconda是一个打包的集合,里面包含了720多个数据科学相关的开源包,在数据可视化、机器学习、深度学习等多方面都有涉及。不仅可以做数据分析,还可以用在大数据和人工智能等领域。安装它后就默认安装了python、IPython、Jupyter notebook和集成开发环境 Spyder等等。
- 并且,可以进行python环境的版本管理(允许用户方便地安装不同版本的python并可以快速切换)
-
安装教程
- 可参考这里的第二部分
- 注意点
-
检查自己电脑有无显卡
- 检查方式:设备管理器 -> 显示适配器 即可看到显卡列表
- 如有显卡,继续进行以下步骤; 如果没有,直接进行第四部分
-
确定安装的CUDA版本:
-
CUDA cuDNN安装
-
python与CUDA对应版本
- 在这里查找对应版本,例如 我安装的是CUDA11.0 那我就应该下载pytorch 1.7.1和torchvision 0.8.2
-
安装方法
- 在线下载也好,离线安装的whl文件也罢,一定一定要记住选择cu版本,而不是cpu版本!安装了cpu版本,torch.cuda.is_available()返回是False。可以通过 pip list 查看自己有没有安装错,错了就pip uninstall,然后进行下列步骤
- 无anaconda利用cmd安装
- pip install torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple some-package(-i 是国内镜像源 直接从官网下有点慢)
- 无anaconda离线安装
- 首先去下载对应版本whl文件,网页上面是cpu版本,记得往下滑,要记住下载的位置
- 记得是两个whl文件 一个torch 一个torchvision
- 打开cmd命令 cd进入该位置 使用pip进行安装
- 有anaconda cmd命令安装
- 使用conda create 命令具体create格式前面应该有涉及,不做赘述
- pip install --target=D:\anaconda\Anaconda3\envs\pytorch\Lib\site-packages torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple some-package (注释:直接使用pip install是安装到系统python中,而我们现在是要安装到虚拟环境中,所以安装指令一定要包含安装路径 --target部分 \pytorch 是我创建的虚拟环境名字 你们要改成对应虚拟环境的名字)
- 有anaconda离线安装
- 结合上面的方法即可实现,不再赘述
- 无anaconda利用cmd安装
- 最后,检验是否安装成功
- 在cmd中输入python
- 输入import torch
- 输入torch.cuda.is_available()
- 如果是True 则说明安装完成
- 在线下载也好,离线安装的whl文件也罢,一定一定要记住选择cu版本,而不是cpu版本!安装了cpu版本,torch.cuda.is_available()返回是False。可以通过 pip list 查看自己有没有安装错,错了就pip uninstall,然后进行下列步骤