配置pytorch环境

 一、安装cuda cudnn

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

配置pytorch环境_第1张图片

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

 2.1下载cuDNN

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

配置pytorch环境_第2张图片

   2.2  配置cuDNN

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

配置pytorch环境_第3张图片

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

配置pytorch环境_第4张图片

二、安装anaconda

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

配置pytorch环境_第5张图片

 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页,根据需要配置选择框内容,最下面生成配置对应执行代码。

配置pytorch环境_第6张图片

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

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

输入y,同意安装

配置pytorch环境_第7张图片

 输入python

       import torch

       torch.cuda.is_avaible()

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

 

你可能感兴趣的:(linux,pytorch,python,深度学习)