linux下torch-0.4.0安装流程及心得

1、https://pytorch.org/：可以按照官网下载对应的whl文件；

如果官网打不开，参考网址https://blog.csdn.net/m0_37615398/article/details/83510891或https://blog.csdn.net/qq_34374211/article/details/80156517的说明安装，注意可以根据自己的硬件平台选择cuda8.0编译的pytorch-0.4.0或者cuda9.0的pytorch-0.4.0；

wget http://download.pytorch.org/whl/cu80/torch-0.4.0-cp36-cp36m-linux_x86_64.whl：wget下载文件

2、https://pypi.org/project/torch/0.4.0/#files网址下也可以下载whl文件，pypi可以搜索下载一些python的第三方库，pytorch-0.4.0也可以在win10下安装，但必须是python3版本；

3、可以参照https://www.jianshu.com/p/131abc0e39a1或https://blog.csdn.net/u010327061/article/details/81001999博客，选择cudnn版本与cuda匹配，例如cuda9.0，cudnn选择7；

4、https://blog.csdn.net/lb1244206405/article/details/90718040博客查看GPU对应的arch型号，在我之前的faster-rcnn的pytorch版本程序运行时编译arch选择可以参见这个网址，例如P100的arch是60，可以选择cuda8及以上版本；V100的arch是70，可以选择cuda9及以上版本；

5、不管是自己安装的环境或者是基于docker的镜像，可以通过cat /usr/local/cuda/version.txt或者nvcc -V查看安装的cuda版本，这里的cuda是自己安装的cuda版本的文件夹，例如安装的cuda9.0，则通过cat /usr/local/cuda-9.0/version.txt查看；一般情况下可以通过cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2查看cudnn版本，同样将cuda换成自己安装的cuda版本目录，如cuda-9.0，不过有时候docker image中cudnn.h不在上述目录下，可以通过find / -name 'cudnn*'查看所有关于cudnn的文件路径，可以看到cudnn.h也可能在/usr/include下，然后查看该目录下的cudnn.h即可；

6、有时候在P100显卡上安装的cuda9.0，cudnn7，但是torch-0.4.0编译选择的是cuda8.0的版本，在P100上能跑，但是V100上就会报错，提示对应的cuda9，但是编译的是8，原因就在于https://blog.csdn.net/lb1244206405/article/details/90718040，P100可以跑cuda8及以上版本，但是V100必须是cuda9及以上，所以需要卸载cuda8.0的torch-0.4.0，安装cuda9.0的torch-0.4.0即可；

pip uninstall torch -y：-y表示直接卸载，不需要提示https://www.jianshu.com/p/728b6956df21

然后pip install  **.whl