conda+py3.7的环境安装torch2.3.1+cuda12.1的版本
时间: 2024-09-18 10:06:16 浏览: 85
conda安装cuda10.1+cudnn7.6.5,conda安装深度学习gpu环境
在Conda环境中安装Py3.7版本的PyTorch 2.3.1以及与其对应CUDA 12.1版本需要按照以下步骤操作:
1. **创建新环境**:
使用Conda创建一个新的环境,命名为`my_pytorch_121`(假设你希望这个环境名字如此):
```bash
conda create -n my_pytorch_121 python=3.7
conda activate my_pytorch_121
```
2. **添加PyTorch和CUDA channels**:
添加PyTorch和CUDA的官方频道,以便安装正确的版本:
```bash
conda config --add channels torch
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch
```
3. **安装PyTorch和CUDA**:
然后安装特定版本的PyTorch和CUDA:
```bash
conda install pytorch torchvision torchaudio cudatoolkit=12.1 -c pytorch -c defaults --override-channels
```
`-c defaults`选项用于默认通道,如果上述渠道无法找到相应的包,可能会用到它。
4. **检查安装**:
安装完成后,确认一下安装情况:
```bash
conda list | grep torch
```
请注意,由于软件包的更新和依赖关系,这一步骤可能需要根据实际的包状态进行调整。如果你遇到问题,可以尝试查找最新的安装指南或从PyTorch官网获取最新信息。
阅读全文