有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_CUDA


关于Anaconda的安装和conda环境的建立,在我的上一篇TF2.0安装过程有写。说实话因为用torch的太多了,所以来学习一手Pytorch安装。

安装之前先废个话,CUDA的安装和cudatoolkit不是一回事,之前初学的时候我以为两个是一个东西,差点出大问题。CUDA只需要在机器上装一个,在虚拟环境里不用。但是如果要TF和torch不同版本并存的话就要找一个结合点,这个方法应该说最简单。

默认装了Visual Studio,没装的装一个去。

默认没装CUDA,先去NVIDIA下个CUDA


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_CUDA_02


现在最新是10.2,但是我们不用,用10.1.243,点击右下角Legacy Releases,出下图:


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_cuda安装失败_03


英文看得懂吧,选个最新的10.1就好了。点击你选的那个出下图


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_虚拟环境_04


先选前三个,最后一个是问你下完整个再本地安装还是网络安装,本地安装包大概2.5个G,网络包才19M,自己看下哪个。我是网络包,下完之后双击EXE,出来下面:


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_cuda安装失败_05


这个路径可以自己选,反正最后C盘里都有。OK以后,先检查系统兼容性, 有问题的就重启试试,这个我也不知道,许可协议也直接点了,然后自定义。


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_cuda安装失败_06


像这样,Experience其实可以不要。CUDA全选上没坏处,但是打开CUDA一项,


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_cuda安装失败_07


如果因为中间的Visual Studio Integration整个安装失败,就先不勾这个,等整个安装完回头再来单独安装它。另外所有的组件,如果电脑有已经安装的版本,且比现在下的这个要安装的更新,就不安装它。这个结束就OK。

但是我说实话看网络上博客,好像都说安装完都要改环境变量,我寻思我也没动手自己改,或者之前装TF的时候改过。如果需要的话就去看看别的博客怎么加的。接下来装torch。

自己先创建一个虚拟环境,我的就叫torch,下面文章也用这个。


进入上面链接,按下图选好


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_CUDA_08


打开anaconda命令行,activate torch以后,输入上图里最后的那个命令,确认安装,等一段时间后安装完成。

接下来安装cudnn。输入:


conda install cudnn=7.6.4


说实话最好用的就这个,英伟达下CUDNN压缩包还要注册,我之前注册过死活登陆不得,用conda也不要复制粘贴文件夹,方便的很。

测试有没有成功。打开虚拟环境,按我下图一样输入,结果一样就OK了。


有pytorch还需要安装libtorch吗 pytorch一定要cuda吗_虚拟环境_09


如果一步步来,应该不会有什么问题。