安装pytorch gpu 1.12.1版本
时间: 2023-11-05 18:01:58 浏览: 227
onnxruntime-gpu-1.12.1-cp37-cp37m-linux-aarch64.whl.zip
以下是安装pytorch gpu 1.12.1版本的步骤:
1. 确认你的GPU支持CUDA 11.1或更高版本,如果不支持,请先更新GPU驱动程序。
2. 安装Anaconda或Miniconda,可以从官方网站下载:https://www.anaconda.com/products/individual 或 https://docs.conda.io/en/latest/miniconda.html
3. 创建一个新的conda环境,命名为“pytorch121”并激活该环境:
```
conda create --name pytorch121 python=3.8
conda activate pytorch121
```
4. 安装pytorch和torchvision:
```
conda install pytorch==1.2.1 torchvision==0.4.2 cudatoolkit=11.1 -c pytorch
```
5. 验证安装是否成功。在Python交互界面中输入以下命令:
```
import torch
print(torch.__version__)
print(torch.cuda.is_available())
```
如果输出的版本号是1.2.1,且CUDA可用,则表示安装成功。
注意:在安装过程中可能会出现依赖项问题,可以尝试使用不同的源,如清华镜像源:
```
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
```
阅读全文