Windows 系统上c++安装PyTorch详细
时间: 2024-06-08 20:09:29 浏览: 34
在 Windows 系统上安装 PyTorch 需要进行以下步骤:
1. 安装 Anaconda:Anaconda 是一个常用的 Python 发行版,包含了众多常用的科学计算库。可以从官网下载并安装:https://www.anaconda.com/products/individual
2. 创建新的 Anaconda 环境:在 Anaconda Prompt 中输入以下命令创建新的环境,并激活该环境。
```
conda create -n pytorch_env python=3.7
conda activate pytorch_env
```
3. 安装 PyTorch:可以使用 pip 或 conda 安装 PyTorch。
使用 pip 安装:
```
pip install torch torchvision
```
使用 conda 安装:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge
```
其中,cudatoolkit=11.1 是指使用 CUDA 11.1 版本进行 GPU 加速。如果没有 GPU,可以省略该参数。
4. 安装 C++ 库:在 Visual Studio 中安装 C++ 开发工具,并在 Anaconda Prompt 中输入以下命令安装相关库。
```
conda install libpython m2w64-toolchain -c msys2
conda install cmake gcc ninja -c conda-forge
```
5. 测试 PyTorch 安装是否成功:在 Python 中输入以下代码测试 PyTorch 是否安装成功。
```python
import torch
print(torch.__version__)
```
如果输出 PyTorch 的版本号,则安装成功。
相关推荐
![doc](https://img-home.csdnimg.cn/images/20210720083327.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)