pytorch环境安装(一次性完成)


前言

换了新电脑,重新安装环境,发现之前遇到的错误都没有再出现了,整个过程很简单。
以下过程都是在安装好anaconda、cuda以及cudnn的环境下进行的。


一、操作步骤

1.创建环境

  1. 直接在anaconda prompt上使用命令行
conda create -n pytorch python=3.8
//-n指的是创建的环境的名字
//python的版本根据自己的电脑进行调整
  1. 也可以如下图所示在图形页面中直接创建一个环境,然后等待下载完毕。
    在这里插入图片描述

2.激活环境并开始安装pytorch

我的cuda版本是11.0(版本号根据自己电脑cuda的版本进行调整),如果遇到问题直接到pytorch官方进行查询你要的命令:

activate pytorch

//使用conda命令
conda install pytorch torchvision cudatoolkit=11.0 -c pytorch
//使用pip命令
pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

//lib torch
Windows binaries do not support Java. Support is only available for Linux and MacOS. Download here for C++ (Release version):
https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-1.6.0.zip

Download here for C++ (Debug version):
https://download.pytorch.org/libtorch/cu102/libtorch-win-shared-with-deps-debug-1.6.0.zip

3.测试pytorch是否安装成功

测试

//输入python进入到python环境中
import torch
//这里可能会报错,需要下载一个组件(https://download.visualstudio.microsoft.com/download/pr/48431a06-59c5-4b63-a102-20b66a521863/4B5890EB1AEFDF8DFA3234B5032147EB90F050C5758A80901B201AE969780107/VC_redist.x64.exe)附网址
import torchvision

//到这里依旧没有问题,安装成功

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值