打算把项目的tensorflow的项目装换成pytorch的,打算在win10电脑上装个环境。
1 安装一个miniconda3,选择miniconda的原因是因为这个方便。安装完成之后,使用命令
1 conda create -n hh
创建虚拟环境,其中上文的中的 hh 是虚拟环境的名称,自己可以随意,叫goudan都可以,python的版本随意,反正后面都会被覆盖。
2 创建环境完成之后,然后为创建的虚拟环境安装pip 工具 conda install pip -n hh ,之所以为虚拟环境安装pip,是因为使用conda安装pytorch,根本装不上,加上*也不行。
我这里是自动给安装上了python3.8
3 安装pytorch,首先进入虚拟环境 conda activate hh ,然后找到pytorch的官网,选择合适的版本,我这里直接给链接了 https://pytorch.org/get-started/locally/ ,我这里使用的是cuda10.2版本的。复制下面的命令 ,然后运行。如果是cpu版本的,CUDA那里直接选择None那一项,安装就好,并且cpu版本的到这里就安装结束,如果只是用来学习,cpu版本的pytorch就够了(顺便说一下,加了*下载速度超快)。
4 安装cuda,最近从英伟达官网下载的cuda安装包都是40多K的不知道什么东西,然后从别人网盘里面找到了安装包,我也把链接贴到这里,我选择的是10.2版本的,里面有10.1,10.2,11三个版本的
链接:https://pan.baidu.com/s/1yD4lyEPHcbzRA6x8krHRpg
提取码:a81b
下载完成之后,一路next,我装到C盘了,如果C盘空间不够,装到别的盘应该也没有问题。
安装完成之后,在命令行中输入 nvcc -V ,有如下输出 ,就表示没有问题。
5 安装cudnn 这个包从官网下载就好,下载的版本要和已经安装的cuda的版本对应,我下载的版本是7.6.5 for CUDA 10.2,
下载完成之后解压缩,进入解压好的文件,将文件夹里面的文件复制到CUDA相应目录文件夹里面就好。 大功告成