当前位置:   article > 正文

PyTorch1.11.0 GPU版本安装(python3.8+pyTorch1.11.0,torch1.11.0+cu113 torchvision0.12.0+cu113)_torch 1.11.0安装

torch 1.11.0安装

目录

一、电脑相关信息

二、安装Pytorch1.11.0 GPU版本

1. 准备:新建虚拟环境

2. 安装pytorch v1.11.0:(在pytorch官网的历史版本里找安装命令)


电脑:win11 + 英伟达显卡(GeForce RTX 3090)+ Anaconda3-2021-05(python3.8.8)+ cuda11.6。

目标:安装gpu版本的PyTorch1.11.0(torch1.11.0/cu113 + torchvision0.12.0/cu113)

写在前面:由于需要安装较新gpu版本的pytorch(感觉显卡越新,cuda版本越高,pytorch版本也需要越新,才能适配),于是再次安装pytorch。电脑里已经装好Anaconda3-2021-05(python3.8.8),电脑里的英伟达3090显卡适配的cuda最高版本为11.6,也已经装好cuda11.6。接下来,开始安装PyTorch1.11.0(torch1.11.0+cu113,torchvision0.12.0+cu113)~~~

一、电脑相关信息

1.  nvidia-smi,可以看到驱动最高支持的cuda版本为11.6(CUDA Version:11.6)。

2. 电脑已经安装好cuda toolkit11.6。

下载地址:CUDA Toolkit Archive | NVIDIA Developer

下载好的exe压缩包(cuda_11.6.2_511.65_windows.exe)如下所示:

点击exe压缩包进行安装,安装路径选择默认的c盘默认路径。安装好之后可以看到:

二、安装Pytorch1.11.0 GPU版本

1. 准备:新建虚拟环境

1)首先安装Anaconda。(很简单,自行搜索)

2)其次在用Anaconda建一个虚拟环境。

  1. conda create -n pytorch1.11.0 python=3.8  (pytorch1.11.0表示自己起的环境的名字。pytorch1.11.0要求python至少是3.7或者3.8,python3.6不行不支持,所以此处不能写python=3.6,此处我选择python=3.8。)
  2. conda activate pytorch1.11.0 (激活虚拟环境)

2. 安装pytorch v1.11.0:(在pytorch官网的历史版本里找安装命令)

1)在上述PyTorch官网历史版本链接( https://pytorch.org/get-started/previous-versions/)里面,找到v1.11.0的安装命令, 此处选择的是cuda11.3对应的PyTorch安装命令。如下图所示:

 

# CUDA 11.3

pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

2)安装pytorch v1.11.0(torch1.11.0+cu1113 ,torchvision0.12.0+cu113)

安装命令:

  1. # CUDA 11.3
  2. pip install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113

3)测试: 

可以看到torch和torchvision对应的版本:torch1.11.0+cu113 ,torchvision0.12.0+cu113

 这样,pytorch v1.11.0 GPU版本(torch1.11.0+cu1113 ,torchvision0.12.0+cu113)就已经安装成功啦!

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/308404?site
推荐阅读
相关标签
  

闽ICP备14008679号