以前我是直接在colab中使用pytorch,非常方便,今天折腾了一上午,终于搞定了pytorh的安装和环境设置,分享下我的安装流程,遇到的问题和解决方案。
1. 用pip安装Pytorch
打开cmd窗口(按win + R),执行下列语句
pip install torch torchvision
执行时先后出现了两个错误:
(1). “ValueError: check_hostname requires server_hostname”
解决方法:关掉VPN
(2). “ERROR: Could not install packages due to an OSError: [WinError 5] 拒绝访问。: 'c:\\python39\\Library'
Consider using the `--user` option or check the permissions.”
问题原因:PIP安装模块的权限不够导致失败
解决方法:加--user
pip install --user torch torchvision
2. 配置环境变量
在安装完PyTorch后,需要将PyTorch的路径添加到系统环境变量中:
在开始菜单中,搜索“系统”:
在系统信息中找到“高级系统设置”,点击进入:
点击“环境变量”,再在弹出来的面板中找到“系统变量”>Path,点击“新建”
将PyTorch安装目录下的“lib”文件夹的路径添加进去
注意具体的路径要根据自己的实际安装路径进行修改,可以在开始菜单中搜索“torch”来找到相关文件夹,寻找相同后缀的地址。
3. 检验是否安装成功,查看版本
import python
出现错误:“ 'import' is not recognized as an internal or external command, operable program or batch file.”
解决方法:先输入python语句,再import torch
然后可以查看torch版本
print(torch.__version__)
(注意“__”是两个短下划线)
4. 在vscode中使用pytorch
注意一定要在vscode中选择安装了torch的环境:
输入:
import torch
但又出现以下问题:
ImportError:failed to load Pytorch C extensions:It appears that Pytorch has loaded the ‘torch/_C’ folder of the Pytorch repository rather than the C extensions which are expected in the ‘torch._C’ namespace.
原因:numpy库重复。环境里本来有一个numpy库,但装torch的时候又附加了另一个版本的numpy库,调用的时候出现调用错乱的问题。
解决方法:在anaconda prompt中激活所用环境,用pip卸载一个numpy。
成功啦!
下面,就可以真正地探索pytorch啦,比如,创建个张量:
x = torch.empty(3, 2)
print(type(x))
print(x)
感觉安装是学习一个库/软件等内容中最难推进的一项了,有时似乎真的是玄学,要不停地试很多种方法,解决了一个问题,又出现了各种各样奇奇怪怪的error,但要相信只要不放弃,总能解决的,祝大家成功!