Ubuntu 下 conda 加速安装pytorch

1.确定自己的CUDA版本

cat /usr/local/cuda/version.txt

2.建立虚拟环境

conda create -n pytorch

开启、退出、删除命令:

source activate pytorch
source deactivate
conda remove -n pytorch --all

3.安装pytorch

https://pytorch.org/ 在官网主页根据你的系统和CUDA,python版本,选择conda安装方式。

使用官方指定的方法下载太慢,所以:

conda config --add channels https://mirrors.
以下是在Ubuntu系统下使用Miniconda安装PyTorch的详细步骤: ### 1. 安装Miniconda 首先,需要从Miniconda的官方网站下载适用于Linux的安装脚本。打开终端,执行以下命令下载安装脚本: ```bash wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh ``` 下载完成后,运行安装脚本: ```bash bash Miniconda3-latest-Linux-x86_64.sh ``` 安装过程中,按照提示进行操作,如阅读许可协议、选择安装路径等。安装完成后,需要初始化Miniconda: ```bash source ~/.bashrc ``` ### 2. 创建并激活新的conda环境 为了避免与其他项目的依赖冲突,建议创建一个新的conda环境来安装PyTorch: ```bash conda create -n pytorch_env python=3.8 ``` 这里创建了一个名为`pytorch_env`的环境,使用Python 3.8版本。创建完成后,激活该环境: ```bash conda activate pytorch_env ``` ### 3. 安装PyTorch 访问PyTorch官方网站(https://pytorch.org/get-started/locally/ ),根据自己的CUDA版本、操作系统等条件选择合适的安装命令。例如,如果使用CPU版本的PyTorch,可以执行以下命令: ```bash conda install pytorch torchvision torchaudio cpuonly -c pytorch ``` 如果系统支持CUDA,且CUDA版本为11.3,可以使用以下命令: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch ``` ### 4. 验证安装 安装完成后,可以在Python中验证PyTorch是否安装成功。在终端中输入`python`进入Python交互环境,然后执行以下代码: ```python import torch print(torch.__version__) ``` 如果没有报错,并且成功打印出PyTorch的版本号,则说明安装成功。 ### 示例代码 以下是一个简单的PyTorch示例代码,用于验证PyTorch是否可以正常运行: ```python import torch # 创建一个张量 x = torch.tensor([1.0, 2.0, 3.0]) y = torch.tensor([4.0, 5.0, 6.0]) # 计算两个张量的和 z = x + y print(z) ```
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值