配置pytorch环境

本文档详细介绍了如何在Windows环境下安装CUDA 11.6和cuDNN,以及如何配置环境变量。接着,通过Anaconda创建并激活名为pytorch的Python 3.9环境,最后指导读者从PyTorch官网获取适用于CUDA 11.6的安装指令,成功安装PyTorch、torchvision和torchaudio,并验证其GPU可用性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

 一、安装cuda cudnn

1.下载cuda ,根据上一步安装pytorch的cuda版本对应,我的是cuda11.6

安装无脑下一步就可以了。

 2.1下载cuDNN

      cuDNN 其实就是 CUDA 的一个补丁而已,专为深度学习运算进行优化的

   2.2  配置cuDNN

解压压缩包,将文件夹里bin、include、lib拷贝到安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6

3. 配置计算机环境变量

将这四个,依次添加到里面

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6\libnvvp

二、安装anaconda

2.1打开anaconda官网   下载软件,详细请参考anaconda简单安装与使用

 2.2配置anaconda环境

打开anaconda prompt

1.创建环境

conda create –n pytorch python=3.9

-n后面的pytorch是你要创建环境的名称,你也可以用其他名称,numpy是你要安装在环境中的包名称。

注意:创建环境时,可以指定要安装在环境中的Python版本。例如创建环境名称为python36,并安装版本的Python3.6在终端中输入:

conda create –n python36 python=3.6

或也可以这样创建环境名称为python27,并安装版本Python2.7:

conda create –n python27 python=2.7

因为项目不同,有时候会用到Python2,还有时候会用到Python3。所以我在自己的计算机上创建了这两个环境,并分别取了这样的环境名称:python27,python36。这样我可以根据不同的项目轻松使用不同版本的python。

2.激活环境

执行:activate pytorch

3. ctrl+D推出环境

三、安装pytorch

打开pytorch官网安装pytorch页,根据需要配置选择框内容,最下面生成配置对应执行代码。

 复制代码,注意执行这里时要以管理员身份,要不没有权限导致失败

conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia

输入y,同意安装

 输入python

       import torch

       torch.cuda.is_avaible()

返回True,表示已经成功安装pytorch。

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值