pytorch安装教程cpu版加上清华源
时间: 2024-12-28 20:32:54 浏览: 13
PyTorch是一个流行的深度学习框架,要在CPU上安装PyTorch,并添加清华大学源以提高下载速度,你可以按照以下步骤操作:
1. **安装Python** (如果尚未安装):
确保已经安装了Python 3.x版本,推荐使用Python 3.6以上。你可以从Python官网下载并安装。
2. **配置环境变量**:
首先需要激活你的Anaconda(如果使用)或创建一个新的虚拟环境。打开命令行(Windows用户可以按Win+R键输入`cmd`),然后输入以下命令(对于新环境):
```
conda create -n myenv python=3.x
conda activate myenv
```
3. **安装pip**:
如果系统默认pip版本较低,可以升级到最新版:
```
pip install --upgrade pip
```
4. **添加清华源**:
在命令行中运行以下命令,将国内镜像源设置为清华大学的PyPI源:
```
pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple/
pip config set global.trusted-host pypi.tuna.tsinghua.edu.cn
```
5. **安装PyTorch**:
使用配置好的pip安装PyTorch CPU版本:
```
pip install torch torchvision cpuonly
```
`cpuonly`参数表示只安装CPU支持的包,避免不必要的依赖。
6. **验证安装**:
安装完成后,可以使用`python -c "import torch; print(torch.__version__)"`检查是否成功安装并查看版本信息。
阅读全文