600字范文,内容丰富有趣,生活中的好帮手!
600字范文 > 做计算机视觉对电脑配置的要求 计算机视觉环境配置

做计算机视觉对电脑配置的要求 计算机视觉环境配置

时间:2023-09-26 17:09:03

相关推荐

做计算机视觉对电脑配置的要求 计算机视觉环境配置

计算机视觉环境配置

1.cuda和cudnn

首先需要根据电脑显卡的版本来选择对应版本的cuda和cudnn。

1)Windows下查看GPU型号,驱动版本,CUDA版本的方法:

命令行输入nvidia-smi

这里的显卡型号为GeForce MX150,驱动版本为441.22,CUDA版本为10.2

2)根据显卡驱动版本号选择对应的CUDA版本号

下载合适版本的CUDA并安装

双击下载好的CUDA安装包进行安装

安装成功后可以看到:

验证是否安装成功:

命令行输入nvcc -V,如果输出如下所示则已正确安装:

3)根据cuda版本号选择对应的cudnn

cuDNN是

解压缩我们下载的CuDnn文件,得到3个文件夹:bin, include, lib。如下图所示,将这个三个文件夹复制到“C:\ProgramData\NVIDIA GPU Computing Toolkit\v8.0”。

2.Anaconda

3.配置Anaconda源

查看当前anaconda源:

conda config --show channels

添加清华镜像源:

conda config --add channels https://mirrors.tuna./anaconda/pkgs/free/

conda config --add channels https://mirrors.tuna./anaconda/pkgs/main/

conda config --set show_channel_urls yes

4.Anaconda虚拟环境

新建虚拟环境:conda create -n your_env_name python=X.X(2.7、3.6等)命令创建python版本为X.X、名字为your_env_name的虚拟环境。

your_env_name文件可以在Anaconda安装目录envs文件下找到

5.python,pytorch和torchvision

根据具体项目需要选择python版本,pytorch版本和torchvision版本

安装anaconda之后,可以使用conda命令在虚拟环境中安装python,pytorch和torchvision

不同系统中安装pytorch,torchvision对应版本命令参照链接

HERE

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。