WIn11+Anaconda 环境下安装 PyTorch

操作系统 Win11,已经预先安装了 Anaconda。

2月10日顺利安装并使用了pyTorch,2月17日import torch时报错:

OSError: [WinError 126] 找不到指定的模块。 Error loading "C:\Users\QHN\anaconda3\lib\site-packages\torch\lib\caffe2_nvrtc.dll" or one of its dependencies.

重新采用虚拟环境方式安装pytortch

1. 为 PyTorch 创建虚拟环境

安装 Anaconda 完毕后,在安装 PyTorch 之前最好先创建一个 pytorch 的虚拟环境。因为 Python 为不同的项目需求创建不同的虚拟环境非常常见。通过创建虚拟环境,相当于为不同的项目创建一块独立的空间,在这个空间里,你安装任何库和框架都是独立的,不会影响到外部环境。参考:WIn10+Anaconda 环境下安装 PyTorch 避坑指南_红色石头Will的博客-CSDN博客

为 PyTorch 单独创建一个空间,也可以分别安装 TensorFlow 了,都创建一个虚拟环境。

创建 PyTorch 虚拟环境步骤:

1.打开 Anaconda 自带的 Anaconda Prompt:

2.打开 Anaconda Prompt 之后,在命令行输入:

conda create --name pytorch python=3.9

注意,此处 pytorch 是虚拟环境的名称,可自行取。

3.输入命令,进入 pytorch 虚拟环境:conda activate pytorch

 好了,pytorch 的虚拟环境创建好了。接下来就是安装 PyTorch 了。

其实,安装 PyTorch 还是非常简单的。接下来记录一些坑。

2. PyTorch 官网安装

PyTorch 的官网提供了简单的安装方法,只需简单的命令即可。

首先,打开 PyTorch 官网安装页面:Start Locally | PyTorch

复制输入命令:conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

我们的环境是:Windows、Conda、Python 3.9、CUDA,

安装完成,测试了一下,导入torch、torchvision都能顺利使用。

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值