Ubuntu安装anaconda和pytorch的几个总结

Ubuntu安装anaconda和pytorch的几个总结

1. conda的几个镜像

清华镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

南大镜像

conda config --add channels http://mirrors.nju.edu.cn/anaconda/pkgs/free
conda config --add channels https://mirrors.nju.edu.cn/anaconda/cloud/conda-forge
conda config --add channels http://mirrors.nju.edu.cn/anaconda/cloud/msys2

2. 删除.condarc中的defaults

这里用conda config --show可以查看相应的安装源信息,但是有个时候却找不到相应的.condarc文件,参考安装anac

<think>嗯,用户想在Ubuntu 20.04上安装CUDAPyTorch。首先,我需要回忆一下相关的步骤。根据之前的引用内容,用户提供了几个参考,其中提到了使用WSL2+Ubuntu20.04+CUDA11.3的安装方法,还有Anaconda安装以及cuDNN的配置。可能需要分步骤来指导用户。 首先,安装CUDA。引用[1]中提到使用runfile(local)安装CUDA 11.3,这样可以避免很多bug。用户需要下载对应的.run文件,然后运行安装命令。应该提醒用户检查系统是否满足要求,比如NVIDIA驱动是否安装,然后禁用nouveau驱动,可能需要重启。安装过程中可能需要选择不安装驱动,如果已经安装了合适的驱动的话。 接下来是cuDNN的安装。引用[2][3]都提到了解压cudnn的文件,并复制头文件库到CUDA的目录下。用户需要先下载cuDNN的tar包,解压后执行复制权限设置命令。这部分需要确保路径正确,尤其是CUDA的安装路径是否为默认的/usr/local/cuda。 然后是Anaconda安装,引用[3]提到下载Anaconda脚本并运行,接着配置环境变量。安装完成后创建conda环境,这样可以在隔离的环境中安装PyTorch,避免依赖冲突。 最后安装PyTorch。引用[2][3]都提到使用conda或pip安装。用户需要去PyTorch官网选择对应的CUDA版本,获取安装命令。例如,对于CUDA 11.3,可能使用conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch安装后需要验证CUDA是否可用,比如运行torch.cuda.is_available()。 可能还需要注意CUDA版本PyTorch版本的兼容性,以及是否安装了正确的cuDNN版本。用户可能会遇到的问题包括驱动不兼容、路径错误或者环境变量未正确设置。另外,使用conda环境可以帮助管理不同的项目依赖。 现在需要将这些步骤结构化,确保用户能够一步步跟随,同时引用正确的参考资料。可能还需要在相关问题里提到常见问题,比如如何验证安装是否成功,如何处理依赖问题,以及如何选择CUDA版本等。</think>### 分步指南:在 Ubuntu 20.04 上安装 CUDA PyTorch #### 一、安装 CUDA 11.3 1. **安装依赖项** ```bash sudo apt update && sudo apt install build-essential ``` 2. **禁用 Nouveau 驱动** 创建文件 `/etc/modprobe.d/blacklist-nouveau.conf`,添加以下内容: ``` blacklist nouveau options nouveau modeset=0 ``` 执行命令后重启: ```bash sudo update-initramfs -u && sudo reboot ``` 3. **下载并安装 CUDA** ```bash wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installers/cuda_11.3.0_465.19.01_linux.run sudo sh cuda_11.3.0_465.19.01_linux.run ``` *安装时取消勾选驱动安装(若已安装NVIDIA驱动)[^1]。* 4. **配置环境变量** 在 `~/.bashrc` 末尾添加: ```bash export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 执行 `source ~/.bashrc` 生效。 --- #### 二、安装 cuDNN 1. **下载 cuDNN 8.2.1(需NVIDIA账号)** 从 [cuDNN Archive](https://developer.nvidia.com/rdp/cudnn-archive) 选择与 CUDA 11.x 兼容的版本[^3]。 2. **安装 cuDNN** ```bash tar -xzvf cudnn-11.3-linux-x64-v8.2.1.32.tgz sudo cp cuda/include/cudnn.h /usr/local/cuda/include/ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* ``` --- #### 三、安装 Anaconda 1. **下载并安装** ```bash wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh bash Anaconda3-2021.05-Linux-x86_64.sh ``` *安装完成后执行 `source ~/.bashrc` 激活环境[^3]。* 2. **创建虚拟环境** ```bash conda create -n pytorch_env python=3.8 conda activate pytorch_env ``` --- #### 四、安装 PyTorch 1. **通过 Conda 安装** ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ``` *或使用 pip(需匹配CUDA版本):* ```bash pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 ``` 2. **验证安装** ```python import torch print(torch.__version__) # 应输出如 1.12.1 print(torch.cuda.is_available()) # 应返回 True ``` --- ###
评论 1
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值