安裝CUDA時需要和Torch版本對應起來,最好先去torch官網上確定要安裝的torch版本對應的CUDA版本。
在安裝CUDA之前需要先確定是否已經安裝驅動,打開終端輸入 nvidia-smi
,若有輸出,則表明驅動安裝過,否則需要先安裝驅動(驅動安裝教程)
一、CUDA11.1安裝
1、CUDA11.1下載
先去CUDA官網上下載要安裝的版本(CUDA11.1下載鏈接),依次選擇Linux——》x86_64——》Ubuntu——》20.04——》runfile(local),根據(jù)自己的電腦配置選擇即可
打開終端,先復制第一條語句到終端下載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接受協(xié)議,繼續(xù)安裝
一般都安裝過驅動了,所以將Driver取消掉,CUDA Toolkit 11.1 是必須要選擇的,其他的可以根據(jù)自己需要進行選擇,選擇完成之后選擇Install進行安裝,接下來等待安裝即可
出現(xiàn)下面內容說明安裝已經完成了
3、環(huán)境配置
在終端中輸入 sudo gedit ~/.bashrc
打開 ~/.bashrc
文件,在最后添加下面語句
export PATH=$PATH:/usr/local/cuda-11.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/lib64
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.1/extras/CUPTI/lib64
保存之后退出,輸入 source ~/.bashrc
進行更新
4、測試CUDA是否安裝成功
在終端中輸入 nvcc -V
,有輸出則代表安裝成功
二、Torch安裝
進入Torch官網(torch官網),根據(jù)自己需要選擇自己要安裝的版本(注意:需要和CUDA版本對應)我安裝的是torch1.10.0版本,在終端中輸入CUDA11.1的安裝命令即可
pip install torch==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的安裝包名稱
根據(jù)提示,按enter繼續(xù)
按q退出許可證查看,輸入yes繼續(xù)按enter確認安裝路徑,等待安裝
輸入yes,通過conda init 啟動conda文章來源:http://www.zghlxwxcb.cn/news/detail-783129.html
四、結束
以上就是CUDA11.1、Torch1.10、Anaconda的安裝過程了,希望對大家有所幫助?。?!文章來源地址http://www.zghlxwxcb.cn/news/detail-783129.html
到了這里,關于【Ubuntu20.04 CUDA11.1+Torch1.10+Anaconda 保姆級安裝教程】的文章就介紹完了。如果您還想了解更多內容,請在右上角搜索TOY模板網以前的文章或繼續(xù)瀏覽下面的相關文章,希望大家以后多多支持TOY模板網!