操作系统 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都能顺利使用。