使用虚拟环境安装GPU版本pytorch成功记录(Win11)


本帖来记录博主试错了无数无数次之后终于成功的办法,在此也推荐在本地下载一直失败/有诸多无法解决的冲突问题的也使用虚拟环境安装pytorch。优点是简单快捷少麻烦,甚至还能解决在官网上下载网络不稳定带来的难题,all in all,虚拟环境完胜。

1、准备步骤

已经安装好anaconda(下载教程从网上找,都很全很保姆)

1.1根据显卡版本下载cuda

查看显卡版本:(win11)在桌面上点击鼠标右键–显示更多选项–进入NVIDIA控制面板
在这里插入图片描述

进入后选择帮助–系统信息
在这里插入图片描述

选择组件,即可在NVCUDA64行的产品名称中中看到可最高选择的cuda版本
!!!!显卡版本要高于cuda版本
在这里插入图片描述

在选好了cuda的版本后,去网上找个教程安装CUDA和CuDNN
!!!!CuDNN的版本要和CUDA对应
为了避免后续的问题,一概推荐从官网下载
CuDNN在下载之后需要把对应的文件放入到CUDA文件夹里,具体操作可以找个有详细步骤的blog

1.2 检查CUDA的安装

怎么检查cuda是否安装成功(以及查看cuda的安装位置)

2.下载带GPU版本的pytorch

在此使用虚拟环境安装pytorch所以不适用常规的直接从官网上conda下载的办法(事实上,由于网络不稳定的问题博主从来没有在这里成功下载过),同时也不推荐清华镜像源的下载,存在自动切换cpu版本的问题,很可能下载了半天发现自己下载来的是CPU版本(吐血)。
在这里插入图片描述

进入上面的install previous versions of PyTorch
根据自己下载的CUDA版本+选择的pytorch版本+win。
Cu开头就是GPU版本,cp后面是python的版本。
比如,我下载的是11.6的CUDA,那么我选择cu116,代码适用于python3.9,那么我的选择如下(python版本选择自由度较高,因为后续创建虚拟环境才会选择为是多少)
在这里插入图片描述

共有三个文件需要下载,torch、torchvision、torchaudio。后两个的选择标准和torch相同。

3.创建虚拟环境

下载好以上三个文件后,在Anaconda prompt中创建新环境

conda create --name d2l python=3.9 -y

d2l是创建的虚拟环境名字,python=3.9是对应的python版本,可以自己指定虚拟环境名字和python版本。!!!!python版本一定要注意,这是与已经安装的torch文件版本对应的。
输入以上代码后,大概等五分钟左右,会得到界面如下
在这里插入图片描述

3.1进入到虚拟环境中

conda activate d2l

将下载的torch三个文件放到虚拟环境的地址
可以看出来,我的虚拟环境地址就是[C:\Users\ASUS],把下载来的whl文件复制到这个地址下
!!!一定要放到虚拟环境的地址,不然会得到如下报错
在这里插入图片描述

3.2下载torch

确认文件所在地址正确以及已经在虚拟环境中运行以下代码(使用conda install不知道为什么会出现和上面一样的报错)

pip install 文件名+whl

将三个文件都要下载
成功界面:
在这里插入图片描述

4.将虚拟环境导入到程序

导入虚拟环境
其中,6、根据目录,依次找到自己创建的虚拟环境路径
描述的不是很准确。可以进入到anaconda中根据创建的环境名字进行查找
在这里插入图片描述

进入后选择environment,就可以看到刚才创建的环境d21,点击名字就可以看到所在的路径,然后根据上面链接中的教程找到目标环境的目录导入python.exe(不是在scripts下面)
在这里插入图片描述

5.确认下载是否成功

选择已经导入的环境作为编译器,创建py文件输入

import torch
print(torch.cuda.is_available())

然后可以看到输出是True,终于成功了!!!!!!!

  • 1
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
### 回答1: 要在Win11安装PyTorch GPU版本,您需要执行以下步骤: 1. 安装CUDA工具包:访问NVIDIA的官方网站并下载适用于您的GPU的CUDA工具包。安装过程中,请确保选择正确的版本和路径。 2. 安装cuDNN:访问NVIDIA的官方网站并下载适用于您的CUDA版本的cuDNN。将cuDNN文件解压缩到CUDA安装目录的相应文件夹中。 3. 安装Anaconda:访问Anaconda的官方网站并下载适用于您的操作系统的Anaconda安装程序。安装过程中,请确保选择正确的版本和路径。 4. 创建虚拟环境:打开Anaconda Prompt并输入以下命令以创建一个新的虚拟环境: conda create --name myenv python=3.8 5. 激活虚拟环境:输入以下命令以激活新创建的虚拟环境: conda activate myenv 6. 安装PyTorch GPU版本:输入以下命令以安装PyTorch GPU版本: conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch 7. 验证安装:输入以下命令以验证PyTorch是否已成功安装python -c "import torch; print(torch.cuda.is_available())" 如果输出为True,则表示PyTorch GPU版本成功安装。 希望这些步骤对您有所帮助! ### 回答2: 首先,PyTorchGPU上的使用是非常方便的,提供了GPU版本安装包,可以提高计算速度。而Windows 11系统相对于Windows 10系统来说,在安装软件方面并没有太大差异,但是在系统的安全和性能上有所提升,更为稳定。因此,安装PyTorch GPU版本Windows 11系统上可以更好地发挥GPU的性能,提高计算效率。 以下是在Windows 11系统上安装PyTorch GPU版本的步骤: 1. 安装CUDA:首先需要在GPU安装CUDA,可以在NVIDIA官网下载最新版本的CUDA,根据要使用GPU型号选择对应的驱动程序和CUDA版本安装好后需要重启电脑。 2. 安装cuDNN:cuDNN是一个用于深度学习GPU加速库,由Nvidia公司开发,可以大幅提高深度学习算法在GPU上的执行速度。在安装完CUDA后,需要在Nvidia开发者网站下载cuDNN库文件,根据CUDA版本选择相应的cuDNN版本并解压缩到CUDA路径下。 3. 安装Anaconda:在官网上下载并安装Anaconda,可以使用conda指令来管理和安装Python项目的依赖库。 4. 创建虚拟环境:在Anaconda中创建一个虚拟环境,可以方便地管理和控制项目的依赖库。 5. 安装PyTorch GPU版本使用conda指令安装PyTorch GPU版本,可以根据需要选择对应的版本和依赖库版本。 6. 测试:安装完成后,在Python环境中导入PyTorch库并创建一个GPU张量,测试PyTorchGPU上的运行情况。 以上就是在Windows 11系统上安装PyTorch GPU版本的步骤,不仅可以提高深度学习算法的速度,还可以充分发挥GPU的性能,提升计算速度。 ### 回答3: 在安装PyTorch GPU版本之前,必须先确保您的计算机满足PyTorch的基本要求:具有32位或64位操作系统,至少有8GB的RAM和一块可以支持CUDA的NVIDIA显卡。 接下来,您需要安装CUDA toolkit,它是一种用于开发和部署GPU加速应用程序的平台。根据您的显卡型号和操作系统,您可以从NVIDIA官网上下载并安装适当版本的CUDA toolkit。 完成CUDA toolkit的安装后,您还需要安装cuDNN(CUDA Deep Neural Network library),它是一种用于深度学习模型的GPU加速库。您需要在cuDNN官网上注册并下载适合您CUDA版本和操作系统的cuDNN库。 安装完CUDA toolkit和cuDNN后,您可以开始安装PyTorch GPU版本。您可以在PyTorch官网上下载适当版本的whl文件并通过pip install命令安装: pip install torch-1.9.0+cu111-cp39-cp39-win_amd64.whl 注意,在安装PyTorch之前,您需要安装numpy和typing_extensions依赖项。 安装PyTorch完成后,您可以验证您的GPU是否被正确识别并可以被PyTorch使用,可以运行以下代码: import torch print(torch.cuda.is_available()) #True print(torch.cuda.get_device_name(torch.cuda.current_device())) #显卡型号 如果输出结果是True,并且显示了您的显卡型号,则说明您已成功安装并配置了PyTorch GPU版本

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值