我安装pytorch用了一个下午和晚上。。。。

安装过程:

目录

安装anaconda

安装pytorch

查看电脑是否支持CUDA:

支持CUDA的电脑:(不支持的自动略过)

anaconda安装pytorch

 创建虚拟环境:

切换到该环境

输入安装命令,命令是在pytorch官网中生成的

测试安装成功:


安装anaconda

网址:Anaconda | Individual Edition

进入之后,下滑到如图,当前有显示合适的python版本,

如果你的和你的python版本不同,则点击下图的 archive,会有一系列的版本适配你的python

然后开始下载安装,安装过程可以看这篇博客 链接:anaconda安装配置教程_振华OPPO的博客世界-CSDN博客_anaconda安装配置

安装pytorch

 去官网下载:本地启动|PyTorch

下滑,找到这个图像,每一行的意思如下:

产品的稳定性:选择产品的稳定性,一般选stable(稳定)

系统:

下载方式:较多的是Conda(需要下载anaconda),pip(一般python就自带)

语言:看你用的语言

CUDA/CPU:如何选择,这要看电脑显卡配置

查看电脑是否支持CUDA:

此电脑右击管理->设备管理器->显示适配器 

 我的电脑就不支持CUDA,所以我选择的就是CPU。

支持CUDA的电脑:(不支持的自动略过)

支持CUDA的显示适配器一般是有两行的,

 那如何查看CUDA的版本:打开CMD,输入nvcc --version

anaconda安装pytorch

在开始找到如下 双击打开:

 创建虚拟环境:

创建名为pytorch,python版本为3.9的环境,环境名字可以自己设置

conda create -n pytorch python=3.9

切换到该环境

conda activate pytorch

  

输入安装命令,命令是在pytorch官网中生成的

conda install pytorch torchvision torchaudio cpuonly -c pytorch

回车后,会需要你选择y/n,选择y,此时开始下载 

当屏幕上显示done,则安装完成 

 

速度可能会比较慢,可以考虑切换源,切换后,安装命令的 -c pytorch则删去

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
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/cloud/msys2/
 
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes

切回默认源: 

conda config --remove-key channels

测试安装成功:

pip list

在刚刚创建的pytorch环境命令行中输入python 进入python,没有反应会是没有显示则成功,然后再输入torch.cuda.is_available(),  安装的是CPU版本的则显示false,CUDA的则显示true

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐