admin管理员组文章数量:1558102
文章目录
- 所需清单
- 1.安装CUDA11.3
- 2.安装cuDNN 11.x
- 3.安装Anaconda
- 4.打开Anaconda Prompt安装配置Python与Pytorch环境
- 5.JupyterNotebook连接Pytorch GPU
所需清单
1. 电脑配置：windows11/10操作系统，RTX3060显卡/其他支持CUDA11.3的型号
2. CUDA 11.3 #具体CUDA版本以Pytorch官网最新推荐为准
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/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
注意：CUDA10.2已不再支持安装Pytorch
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
#GPU加速
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()
c=torch.matmul(a,b)
t2=time.time()
print(a.device,t2-t0,c.norm(2))
输出以上结果则安装成功！
版权声明：本文标题：Win11-RTX3060-配置Pytorch GPU 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/xitong/1725845388a1045346.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。
发表评论