PyTorch - 安装

---

前往官方下载界面，选择自己的操作系统和安装方式，可以得到安装命令：

https://pytorch.org/get-started/locally/

这里我们得到的命令

pip install torch torchvision torchaudio

---

由于一般下载比较慢，这里我使用豆瓣的源

pip install torch torchvision torchaudio  -i https://pypi.douban.com/simple/

---

import torch
torch.__version__  # '1.7.1'

---

---

关于对 macOS 的支持：

PyTorch 支持 macOS 10.10 以上系统；
建议但不是必须 使用 NVIDIA GPU 来使用 CUDA 支持；

• mac 外接显卡实现深度学习
  https://blog.csdn.net/jiujiu31415/article/details/86485357

---

测试

import torch
x = torch.rand(5, 3)
print(x)
'''
tensor([[0.0075, 0.7693, 0.5815],
        [0.9248, 0.5054, 0.0352],
        [0.3801, 0.5588, 0.2386],
        [0.6712, 0.4793, 0.9562],
        [0.3126, 0.6879, 0.9574]])
'''

---

关于显卡

深度学习一般需要 NVIDIA 的显卡，一般是 960，970，1060，1070；
预算不够可以购买 GTX 1050TI 4G，用于学习是足够的；
预算足够的话，可以买 GTX 1080TI。

win 上检验cuda 安装：

$ mvcc -V 

如果 mvcc 不识别

---

windows 安装：
https://blog.csdn.net/zazazaz1/article/details/108085918