欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

windows10配置pytorch

程序员文章站 2022-06-12 09:14:09
...

1、下载Anaconda,从官网下

2、下载cuda并安装

以下操作需要打开cmd,输入指令运行:

3、创建虚拟环境,-n 后面pytorch为虚拟环境的名字以及python的版本,根据自己情况修改:

 conda create -n pytorch python=3.6

 4、**虚拟环境,安装pytorch

conda activate pytorch

5、从pytorch官网配置在的版本

windows10配置pytorch

然后run:

pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

加豆瓣镜像,下载会很快。所以,最后运行命令就如下:

pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i http://pypi.douban.com/simple --trusted-host pypi.douban.com