【pytorch】Ubuntu18.04.1安装pytorch

安装Anaconda

本部分参考自此博客
由于Anaconda在处理多个虚拟环境方面比较方便,所以还是先安装Anaconda,进入终端后直接使用命令下载即可

  1. 中科大镜像下载(推荐)
    wget https://mirrors.ustc.edu.cn/anaconda/archive/Anaconda3-5.2.0-Linux-x86_64.sh
    
  2. 官网下载
    wget https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.sh
    
    下载完Anaconda3-5.2.0-Linux-x86_64.sh,执行
    bash Anaconda3-5.2.0-Linux-x86_64.sh
    
    安装过程中会有许多需要确认的地方,根据提示进行操作即可。默认安装路径即为Anaconda3-5.2.0-Linux-x86_64.sh所在路径,中途会问你是否需要加入环境变量,选择即可
    注:根据我个人安装的经验,即使在安装过程中已经允许加入环境变量,但是安装进程结束后,使用conda却提示报错,此时需要执行exp
### Ubuntu系统中通过Conda安装PyTorchUbuntu系统中,使用Conda来管理Python环境并安装PyTorch是一种高效的方式。这不仅简化了依赖项管理和隔离不同项目的需求,还使得切换不同的PyTorch版本变得容易。 #### 创建新的Conda环境 为了保持项目的独立性和整洁性,在安装之前建议先创建一个新的Conda环境: ```bash conda create --name my_pytorch_env python=3.9 ``` 激活新创建的环境以便后续操作在此环境中执行: ```bash conda activate my_pytorch_env ``` #### 安装带有CUDA支持的PyTorch (GPU版) 对于希望利用图形处理单元(GPU)加速计算的任务来说,可以按照官方推荐的方法安装具有CUDA支持的PyTorch版本[^2]。确保已正确配置好NVIDIA驱动程序以及相应的CUDA工具包之后,可以通过下面这条命令完成安装: ```bash conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia ``` 此命令会从`pytorch`和`nvidia`两个渠道下载所需的软件包,并自动解决任何潜在的依赖关系问题[^3]。 #### 安装仅限于中央处理器(CPU)使用的PyTorch 如果目标机器不具备合适的GPU硬件条件,则可以选择只针对CPU优化过的PyTorch发行版。这种方式同样能够满足许多应用场景下的需求,特别是当数据集规模不大时。对应的安装指令如下所示: ```bash conda install pytorch torchvision torchaudio cpuonly -c pytorch ``` 上述两条路径均能有效部署适合特定工作负载特性的PyTorch运行环境,具体选择取决于实际的应用场景和技术栈要求。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值