window 清华源安装pytorch11.6
时间: 2023-05-08 09:59:06 浏览: 622
清华镜像源,用于帮助快速下载pytorch
要在Window操作系统上安装PyTorch 11.6,首先需要在清华镜像源中下载安装包。在此之前,需要在自己的电脑上安装conda环境,以便使用conda来管理python的安装及PyTorch的包。
第一步,打开清华镜像源的网站(http://mirrors.tuna.tsinghua.edu.cn/),在搜索框中搜索“pytorch 1.6”,选择与自己电脑系统匹配的安装包进行下载。
第二步,打开anaconda prompt或者powershell,输入以下命令创建一个名为“PyTorch”环境:
```
conda create -n PyTorch python=3.7
```
其中,name可以自己定义,python版本需要根据安装包的系统要求进行选择。
第三步,激活创建的环境并安装PyTorch及torchvision,具体命令如下:
```
conda activate PyTorch
conda install pytorch=1.6.0 torchvision cudatoolkit=10.2 -c pytorch
```
其中,cudatoolkit的版本号需要根据自己电脑上的GPU类型确定,如果没有GPU则可以省略该参数。
最后,输入以下命令测试是否安装成功:
```
python
import torch
torch.cuda.is_available()
```
如果输出结果为True,则表明安装完成且成功使用GPU加速。如果输出结果为False,则可能是安装过程中出现了问题,需要进行排查。
阅读全文