当前位置:   article > 正文

简单快速的cuda11.7配置及torch2.0安装

cuda11.7

一、cuda环境配置

1. cuda下载-https://developer.nvidia.com/cuda-toolkit-archive

在这里插入图片描述
在这里插入图片描述

2. 下载cuda对应的cudnn-https://developer.nvidia.com/rdp/cudnn-archive

在这里插入图片描述

3.对cuda安装过程【无脑安装-一律按是】-C盘program files文件中会出现NVIDIA GPU Computing Toolkit

在这里插入图片描述
在这里插入图片描述

4.将cuda对应的cudnn的文件下的bin、include、lib包分别复制对应的cuda文件夹下的bin、include、lib包中

例如-cudnn中的bin包文件,全部复制到cuda中的bin中
在这里插入图片描述
在这里插入图片描述

复制到cuda中的bin文件夹里

5.nvidia-smi检验

在这里插入图片描述

二、anaconda里建立环境

1.创建环境

在这里插入图片描述

conda create -n AI_2 python=3.8

activate AI_2
  • 1
  • 2
  • 3

2.进入pytorch官网页面-https://pytorch.org/get-started/locally/ 复制下载命令

在这里插入图片描述

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117```


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

闽ICP备14008679号