pytorch2.0.1对应conda什么版本
时间: 2023-12-06 21:05:07 浏览: 202
1.10.0GPU版本pytorch环境,conda安装支持cuda10.2和11.3 pip安装支持cuda10.2和11.1
根据PyTorch官网的说明,PyTorch 2.0.1需要使用conda 4.9.0及以上版本。你可以使用以下命令安装conda 4.9.0:
```
wget https://repo.anaconda.com/archive/Anaconda3-2020.11-Linux-x86_64.sh
bash Anaconda3-2020.11-Linux-x86_64.sh
```
安装完毕后,你可以使用以下命令创建一个名为pytorch的虚拟环境,并安装PyTorch 2.0.1:
```
conda create -n pytorch python=3.7
conda activate pytorch
conda install pytorch==2.0.1 torchvision==0.8.2 cpuonly -c pytorch
```
这里安装的是CPU版本的PyTorch,如果你想要安装GPU版本的PyTorch,可以使用以下命令:
```
conda install pytorch==2.0.1 torchvision==0.8.2 cudatoolkit=10.2 -c pytorch
```
注意:如果你的机器上没有安装NVIDIA的GPU驱动程序和CUDA,请先安装它们。
阅读全文