安装CUDA时需要和Torch版本对应起来,最好先去torch官网上确定要安装的torch版本对应的CUDA版本。
在安装CUDA之前需要先确定是否已经安装驱动,打开终端输入
nvidia-smi
,若有输出,则表明驱动安装过,否则需要先安装驱动(驱动安装教程)
一、CUDA11.1安装
1、CUDA11.1下载
先去CUDA官网上下载要安装的版本(CUDA11.1下载链接),依次选择Linux——》x86_64——》Ubuntu——》20.04——》runfile(local),根据自己的电脑配置选择即可
打开终端,先复制第一条语句到终端下载CUDA
wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run
下载完成之后,输入第二句语句进行安装
sudo sh cuda_11.1.0_455.23.05_linux.run
2、安装过程
选择Continue,按enter键进行下一步
输入accept接受协议,继续安装
一般都安装过驱动了,所以将Driver取消掉,CUDA Toolkit 11.1 是必须要选择的,其他的可以根据自己需要进行选择,选择完成之后选择Install进行安装,接下来等待安装即可
出现下面内容说明安装已经完成了
3、环境配置
在终端中输入
sudo gedit ~/.bashrc
打开
~/.bashrc
文件,在最后添加下面语句
exportPATH=$PATH:/usr/local/cuda-11.1/bin
exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
exportLD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/extras/CUPTI/lib64
保存之后退出,输入
source ~/.bashrc
进行更新
4、测试CUDA是否安装成功
在终端中输入
nvcc -V
,有输出则代表安装成功
二、Torch安装
进入Torch官网(torch官网),根据自己需要选择自己要安装的版本(注意:需要和CUDA版本对应)
我安装的是torch1.10.0版本,在终端中输入CUDA11.1的安装命令即可
pip installtorch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
三、Anaconda安装
1、Anaconda下载
进入Anaconda的官网,点击Download进行下载
如果官网下载太慢了,可以从清华源来下载(Anaconda镜像地址),选择最近这两年的版本下载
2、Anaconda安装
在终端中打开Anaconda的下载目录,输入
bash Anaconda3-2022.05-Linux-x86_64.sh
,bash 后面为Anaconda的安装包名称
根据提示,按enter继续
按q退出许可证查看,输入yes继续
按enter确认安装路径,等待安装
输入yes,通过conda init 启动conda
四、结束
以上就是CUDA11.1、Torch1.10、Anaconda的安装过程了,希望对大家有所帮助!!!
版权归原作者 尽量不拖延的小王 所有, 如有侵权,请联系我们删除。