Ubuntu18.04+CUDA10.1

因为我要跑affnet,需要的环境是：
python3.7,pytorch1.4.0

经过查询，cuda及其可用PyTorch对应版本（参考官网，欢迎评论区补充）：

需要安装10.1版本的CUDA

各个版本的CUDA下载网站：
https://developer.nvidia.com/cuda-toolkit-archive

因为我要下载10.1版本的：

zhenghan@zhenghan:~/Software$ sudo sh cuda_10.1.105_418.39_linux.run
[sudo] zhenghan 的密码： 
===========
= Summary =
===========

Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-10.1/
Samples:  Installed in /home/zhenghan/, but missing recommended libraries

Please make sure that
 -   PATH includes /usr/local/cuda-10.1/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-10.1/lib64, or, add /usr/local/cuda-10.1/lib64 to /etc/ld.so.conf and run ldconfig as root

To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-10.1/bin

Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-10.1/doc/pdf for detailed information on setting up CUDA.
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least 418.00 is required for CUDA 10.1 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:
    sudo <CudaInstaller>.run --silent --driver

Logfile is /var/log/cuda-installer.log
zhenghan@zhenghan:~/Software$ sudo vim ~/.bashrc
zhenghan@zhenghan:~/Software$ source ~/.bashrc
zhenghan@zhenghan:~/Software$ sudo rm -rf /usr/local/cuda  
zhenghan@zhenghan:~/Software$ sudo ln -s /usr/local/cuda-10.1 /usr/local/cuda  
zhenghan@zhenghan:~/Software$ nvcc --version  
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105

注意：如果有显卡，安装过程中千万不要选驱动：、

查看CUDA版本，nvcc -V

zhenghan@zhenghan:/$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105
zhenghan@zhenghan:/$ 

至此，CUDA10.1安装成功！！