我安装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