使用Anaconda下载安装Pytorch

Pytorch的安装,常用命令

  • 虚拟环境操作
  • 镜像源操作:
  • 更改pip下载路径
    • 找到site.py文件
    • 更改USER_SITE 、USER_BASE
    • 验证
  • 查看cuda版本号
  • pt_gpu下载安装,适合cuda11.6

虚拟环境操作

操作 语句
查看 conda env list
创建 conda create -n <环境名> python=<版本号>
删除 conda env remove -n <环境名>
激活 conda activate <环境名>
退出 conda deactivate

镜像源操作:

操作 语句
查看 conda config --show channels
添加 conda config --add channels
搜索时显示通道地址 conda config --set show_channel_urls yes
删除 conda config --remove channels
重置 conda config --remove-key channels

更改pip下载路径

pip默认下载位置在C盘,如果需要更改位置可以参考下列方法

找到site.py文件

打开命令提示符(win+r),输入cmd,输入命令: python -m site -help
使用Anaconda下载安装Pytorch_第1张图片
用记事本打开site.py文件

更改USER_SITE 、USER_BASE

使用Anaconda下载安装Pytorch_第2张图片

验证

输入命令:python -m site

查看cuda版本号

打开cmd,输入命令:nvidia-smi
使用Anaconda下载安装Pytorch_第3张图片

pt_gpu下载安装,适合cuda11.6

创建虚拟环境:conda create -n pt_gpu python=3.7
激活虚拟环境:conda activate pt_gpu
下载相关库1pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116
验证:

python

import torch

torch.cuda.is_available()

显示Ture即可


  1. pytorch_gpu下载安装,适合cuda11.6,极简代码风 ↩︎

你可能感兴趣的:(pytorch,python,深度学习)