所需清单
1. 电脑配置:windows11/10操作系统,GTX3060显卡
2. CUDA 11.3
3. cuDNN 11.x
4. Anaconda
5. 镜像源安装Pytorch
6. JupyterNotebook测试程序
1.安装CUDA11.3 登录英伟达官网找到CUDA11.3下载链接并下载安装CUDA 11.3版本
注意项: 1.若已经安装过CUDA,先删除掉已安装的CUDA历史版本的整个文件夹 一般默认的CUDA目录位于:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\ 2.安装CUDA 11.3时不要勾选CUDA中的Visual Studio Intergration 3.通过win+R输入control system 进入系统高级设置->环境变量,查看环境变量是否添加成功 系统变量path中要有 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin
与C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\libnvvp
2.安装cuDNN 11.x 同样在英伟达官网中搜索cuDNN[cuDNN下载链接](https://developer.nvidia.com/rdp/cudnn-download) 并下载11.x版本, 将解压后的所有文件复制替换至C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3 的同名文件 3.安装Anaconda 具体步骤自行搜索 4.打开Anaconda Prompt安装配置Python与Pytorch环境 具体步骤参考总结配置pytorch GPU环境的两种方法中的三:安装PyTorch 区别在于将最后一句安装命令中的CUDA 10.2版本更换为11.3 5.JupyterNotebook连接Pytorch GPU 重新打开Anaconda Prompt,base环境下输入conda install nb_conda_kernels 再进入配置好的torch环境输入conda install nb_conda 最终输入jupyter notebook运行打开进行测试
import torch
import time
from torch import autograd
print(torch.__version__)
print(torch.cuda.is_available())
a=torch.randn(10000,1000)
b=torch.randn(1000,10000)
print(a)
print(b)
t0=time.time()
c=torch.matmul(a,b)
t1=time.time()
print(a.device,t1-t0,c.norm(2))
device=torch.device('cuda')
print(device)
a=a.to(device)
b=b.to(device)
t0=time.time()
c=torch.matmul(a,b)
t2=time.time()
print(a.device,t2-t0,c.norm(2))
t0=time.time()
c=torch.matmul(a,b)
t2=time.time()
print(a.device,t2-t0,c.norm(2))
t1=time.time()
print(a.device,t1-t0,c.norm(2))
device=torch.device('cuda')
print(device)
a=a.to(device)
b=b.to(device)
t0=time.time()
c=torch.matmul(a,b)
t2=time.time()
print(a.device,t2-t0,c.norm(2))
t0=time.time()![在这里插入图片描述](https://img-blog.csdnimg.cn/915baca0ee9e43ec9b4d971411fd9659.png?x-oss-process=image/watermark,type_d3F5LXplbmhlaQ,shadow_50,text_Q1NETiBA6bmF5q-b5Zyo6Lev5LiK5LqG,size_20,color_FFFFFF,t_70,g_se,x_16)
c=torch.matmul(a,b)
t2=time.time()
print(a.device,t2-t0,c.norm(2))
输出以上结果则安装成功!
|