PyTorch安装 -Python深度学习模块-各种问题
打开网址https://pytorch.org/get-started/locally/
找到对应的版本
由于我的CUDA是10.1版本 所以我应该选择“none”(选择对应的cuda很重要)
由于Anoconda安装网络下载经常会出现各种各样的失误
所以我们都选择用pip 安装。
加粗样式
打开王也
打开这个网页下载whl文件,放到
这个文件夹里,然后
重要的步骤来了
1、打开Anoconda promt
pip3 install 下载文件名称
pip3 install torchvision
此时会发现只有在anoconda里面 是可以导入torch模块的 ,但是在IDLE中不能用 ,因此 接下来继续同样安装
2、通过cmd打开命令窗口
pip3 install 下载文件名称
pip3 install torchvision
同样进行安装
3、打开pycharm等待更新
即可以看到可以导入
4、测试代码
import torch
x1=torch.rand(5,3)
print(x1)
x2=torch.rand(5,3)
print(x2)
x3=x1+x2
print(x3)
输出是:
tensor([[0.9286, 0.6405, 0.4798],
[0.6246, 0.2197, 0.6535],
[0.5857, 0.0235, 0.5057],
[0.3193, 0.9208, 0.6338],
[0.8803, 0.2905, 0.9775]])
tensor([[0.7838, 0.0902, 0.5741],
[0.3190, 0.4968, 0.3553],
[0.8352, 0.1472, 0.8644],
[0.7145, 0.4298, 0.0411],
[0.8267, 0.5461, 0.3833]])
tensor([[1.7124, 0.7307, 1.0539],
[0.9436, 0.7165, 1.0088],
[1.4209, 0.1707, 1.3701],
[1.0339, 1.3506, 0.6749],
[1.7070, 0.8366, 1.3608]])
即可说明 成功安装
如果不好使,可以卸载
下载后 通过步骤重新安装