栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

2021-10-09

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

2021-10-09

                             CUDA、Anaconda和pytorch的环境安装及测试是否安装成功

一.安装CUDA

        首先,查看自己的电脑的cuda的版本。打开NVIDIA的控制面板 从帮助 - 系统信息 - 组件 - NVCUDA64.DLL 查看你显卡驱动能支持的最高cuda版本。

 然后下载对应安装包。我的是11.1版本

然后进行安装,建议无脑安装,尽量不要修改它默认的存储路径。 安装完成之后,因为是精简安装,会默认安装GeForce Experience,在GeForce Experience中可以更新你的显卡驱动(可以支持版本更高的CUDA)。中途可能会遇到报错,这时就选择自定义,继续下一步,就会弹出这个去掉报错文件,继续安装。当然也可能是你之前安装了一些其他软件,我之前安装了VScommunity,下载了VS2019,就有文件安装失败,如果你卸载了它,那就能正常安装CUDA,把它之后再重新安装。

 

 

二.安装Anaconda

        首先,下载Anaconda的安装包。

 

 

         记得下载符合你操作系统的版本, 是32位就下32位, 是64位就下64位,是MAC就下MAC,是linux64就下linux64,这个搞错挺丢人的(我就搞错了)。

         接着,进行安装。

 

 

 记得如果用户和我一样搞错了名字加了非法字符的,在另一个盘重开一个文件夹进行安装。接着一直Next就行了。

       接着打开

 出现这个界面

 看到这里的(base)了吗?是安装了anaconda后的默认环境,我们需要重新搞一个

输入conda create -n your_name python=3.7。这里create就是创建一个新的虚拟环境环境 -n 就是 -name的缩写 your_name 可以替换成你们自己取的环境的名字 python=3.7 你也可以指定不同的python版本。

接着就会出现

 

 输入y,再回车即可

 activate  your_name 就是激活你刚刚创建的环境,激活完之后,(base)就会变成(your_name)。输入conda env list 是列出你所有创建的环境。

三.安装pytorch

       进入https://pytorch.org/get-started/locally/网站,下载符合你的版本

        记得把整个字符都复制下来比如我是这个pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f

然后按下回车,即可安装。

四.进行测试。

         首先,激活你的虚拟环境,然后输入python

 

          判断是否安装了cuda import torch print(torch.cuda.is_available())  返回True,cuda就安装好了。

           判断是否安装了cuDNN from torch.backends import  cudnn print(cudnn.is_available()) 返回True,cudnn就安装好了。

           之后可以运行一段代码来感受一下所安装cuda的效果

import torch
import time

a = torch.randn(400,600,600).to("cuda:0")

b = torch.randn(400,600,600).to("cuda:0")

start_time = time.time()

for i in range(1,1000):
    c = a*b

end_time = time.time()

print("CUDA time: ",(end_time-start_time))


a = torch.randn(400,600,600)

b = torch.randn(400,600,600)

start_time = time.time()

for i in range(1,1000):
    c = a*b

end_time = time.time()


print("CPU time: ",(end_time-start_time))

         第一次写,如有不足,请多包涵。 

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/308968.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号