简单快速的cuda11.7配置及torch2.0安装
-
- 一、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包中
- 5.nvidia-smi检验
- 二、anaconda里建立环境
- 1.创建环境
- 2.进入pytorch官网页面-https://pytorch.org/get-started/locally/ 复制下载命令
一、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中
5.nvidia-smi检验
二、anaconda里建立环境
1.创建环境
conda create -n AI_2 python=3.8
activate AI_2
2.进入pytorch官网页面-https://pytorch.org/get-started/locally/ 复制下载命令
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117```