pytorch1.13.0 cuda11.6安装清华源
时间: 2024-11-25 22:11:48 浏览: 1
深度学习环境搭建pytorch安装教程
5星 · 资源好评率100%
PyTorch 1.13.0 和 CUDA 11.6 的安装通常需要设置国内的清华源,以便加速包的下载速度。以下是安装步骤:
1. **安装前准备**:
- 首先确保已安装了Python 3.6及以上版本。
- 安装`conda`环境管理器(如果你还没有安装),可以访问 https://docs.conda.io/en/latest/miniconda.html 下载并安装。
2. **创建conda环境**:
```
conda create -n pytorch_env python=3.7 # 创建一个名为pytorch_env的新环境
conda activate pytorch_env # 激活新环境
```
3. **配置清华源**:
```
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud # 添加清华大学源
conda config --set channel_priority strict # 设置默认优先级为严格
```
4. **安装PyTorch**:
```
conda install pytorch torchvision cudatoolkit=11.6 -c pytorch # 使用清华源安装PyTorch 1.13.0和对应CUDA版本
```
如果需要特定版本,替换`1.13.0`为所需版本,如`pytorch=1.13.0 torchvision=0.13.0 cudatoolkit=11.6`.
5. **检查安装**:
```bash
python -m torch.cuda.is_available() # 确保GPU支持可用
```
记得每次激活这个环境时,都要使用`conda activate pytorch_env`命令。
阅读全文