win10安装pytorch-gpu

步骤总结:

1. 下载cuda toolkit11.3, 完成环境配置path,安装后cmd窗口输入nvcc -V查看安装信息

2. 下载cudaa 11.x, 解压缩到cuda11.3覆盖三个文件夹

3. conda create -n mytorch python=3.7 创建新环境,装过cpu版的一定要在新环境中重装.

4. 按照其他教程安装验证为True的就结束。这里写False的解决方法,针对conda list查看pytorch是cpu版本的情况(可能和源有关)——

https://download.pytorch.org/whl/torch_stable.html手工下载电脑对应版本的三个whl包,pytorch官网自动显示的命令也是这三个包:

使用pip install 路径\文件名  分别进行安装

5. pip install jupyter ,下载jupyter notebook到新环境,命令窗口输入jupyter notebook启动

 win10安装pytorch-gpu_第1张图片

你可能感兴趣的:(pytorch,人工智能,python)