赞
踩
参考文档
https://datawhalechina.github.io/thorough-pytorch/%E7%AC%AC%E4%B8%80%E7%AB%A0/1.2%20PyTorch%E7%9A%84%E5%AE%89%E8%A3%85.html
1:安装Anaconda
登录Anaconda | Individual Edition,https://www.anaconda.com/download/success ,选择相应系统DownLoad,此处以ubuntu20.0.4为例,
下载的安装文件为Anaconda3-2024.02-1-Linux-x86_64.sh,安装之。
2,创建虚拟环境,本文的python版本为3.11.7,环境的名称为nbpytorch
conda create -n nbpytorch python==3.11.7
3,安装包,本文安装的包为pandas
conda install pandas
4,激活环境命令, 本文的环境的名称为nbpytorch
conda activate nbpytorch
5,其他命令
(i),卸载包
conda remove package_name
# 注:package_name 替换成对应的包的名称,eg: pandas
(ii),显示所有安装的包
conda list
(iii),删除虚拟环境命令
conda remove -n env_name --all
# 注:env_name 替换成对应的环境的名称
(iv),退出当前环境
conda deactivate
6,安装PyTorch
(1):登录PyTorch官网 https://pytorch.org/
选择配置,本文的pytorch安装命令如下,在终端上执行:
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
(2), 在终端上执行激活环境(如果已激活,本步骤省略)
conda activate nbpytorch
(3),检验是否安装成功
在上一步已激活环境的情况下,在终端执行命令python,然后输入下面的代码。
import torch
torch.cuda.is_available()
这条命令意思是检验是否可以调用cuda,如果我们安装的是CPU版本的话会返回False,
能够调用GPU的会返回True。一般这个命令不报错的话就证明安装成功。
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。