使用Anaconda下载安装Pytorch

文章详细介绍了如何在Windows环境下使用conda管理虚拟空间,安装指定版本的Python,并针对CUDA11.6配置Pytorch,包括更改pip下载路径、检查CUDA版本以及验证GPU支持。步骤包括创建和管理conda环境、设置镜像源、安装Pytorch及其依赖库torchvision和torchaudio。
摘要由CSDN通过智能技术生成

虚拟环境操作

操作语句
查看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
site.py文件位置
用记事本打开site.py文件

更改USER_SITE 、USER_BASE

在这里插入图片描述

验证

输入命令:python -m site

查看cuda版本号

打开cmd,输入命令:nvidia-smi
在这里插入图片描述

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,极简代码风 ↩︎

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值