linux系统pytorch清华源下载
时间: 2024-09-10 15:24:14 浏览: 87
在Linux系统上安装PyTorch时,通常推荐使用Anaconda这样的科学计算环境管理工具,尤其是当你需要访问清华大学的镜像源时,因为这可以加速软件包的下载速度。以下是使用Anaconda安装PyTorch的基本步骤:
1. **安装Miniconda**:
- 访问官网:https://www.anaconda.com/products/miniconda
- 下载适合Linux的Miniconda,并按照指示安装。
2. **激活Conda环境**:
打开终端,运行 `conda init` 添加Conda命令行支持,然后激活一个新的环境,例如创建一个名为`pytorch_env`的新环境:
```
conda create -n pytorch_env python=3.7
conda activate pytorch_env
```
3. **添加清华源**:
```
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/
conda config --set channel_priority strict
```
4. **安装PyTorch**:
使用Conda安装PyTorch,比如安装CPU版本:
```
conda install pytorch torchvision cudatoolkit=10.2 -c anaconda -c pytorch
```
如果你需要GPU支持,记得检查你的显卡兼容性和对应的CUDA版本。
5. **验证安装**:
完成安装后,你可以通过 `python -m torch` 或者 `python -c "import torch; print(torch.__version__)"` 来确认PyTorch是否成功安装并查看版本。
阅读全文