pycharm 安装pytorch (pip安装)
接下来只需要找到并下载对应的库就行了。指令里面库的命名方式和库名称并不一样,但是包含了相应的信息。在Windows下对应关系为:
torch==1.8.0+cu102对应 cu102/torch-1.8.0-cp37-cp37m-win_amd64.whl
其中cu102对应cu102,torch==1.8.0对应torch-1.8.0,cp37代表Python版本,我这里是3.7,后面对应系统版本;以此类推。
torchvision==0.9.0+cu102对应 cu102/torchvision-0.9.0-cp37-cp37m-win_amd64.whl
torchaudio===0.8.0 对应 torchaudio-0.8.0-cp37-none-win_amd64.whl
下载安装相应的库就可以了。
而且我发现这样下载速度还很快,上M了,也是神奇~
装好后在pycharm输入:
import torch
x = torch.rand(5, 3)
print(x)
对应输出为:
tensor([[0.8236, 0.8568, 0.0074],
[0.1060, 0.1674, 0.8331],
[0.9277, 0.9373, 0.5105],
[0.5573, 0.6022, 0.0768],
[0.6844, 0.1197, 0.5525]])
搞定!