赞
踩
目录
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盘默认路径。安装好之后可以看到:
1)首先安装Anaconda。(很简单,自行搜索)
2)其次在用Anaconda建一个虚拟环境。
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)
安装命令:
- # 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
3)测试:
可以看到torch和torchvision对应的版本:torch1.11.0+cu113 ,torchvision0.12.0+cu113
这样,pytorch v1.11.0 GPU版本(torch1.11.0+cu1113 ,torchvision0.12.0+cu113)就已经安装成功啦!
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。