88人参与 • 2024-08-06 • 智能机器人
安装cuda时需要和torch版本对应起来,最好先去torch官网上确定要安装的torch版本对应的cuda版本。
在安装cuda之前需要先确定是否已经安装驱动,打开终端输入 nvidia-smi
,若有输出,则表明驱动安装过,否则需要先安装驱动()
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
文件,在最后添加下面语句
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官网),根据自己需要选择自己要安装的版本(注意:需要和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
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的安装过程了,希望对大家有所帮助!!!
您想发表意见!!点此发布评论
版权声明:本文内容由互联网用户贡献,该文观点仅代表作者本人。本站仅提供信息存储服务,不拥有所有权,不承担相关法律责任。 如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 2386932994@qq.com 举报,一经查实将立刻删除。
发表评论