GPU:NVIDIA GeForce GTX 1050
Anaconda3-2020.11-Windows-x86_64(已安装)
对应python3.8.5
PyTorch安装
打开Anaconda Prompt,初始状态是(base)开头
创建环境
命令:conda create -n (环境命名) python=(版本)
将会在C:\Users\18210\anaconda3\envs里建立
本次输入命令:conda create -n pytorch python=3.8
进入环境
本次输入命令:conda activate pytorch
此时变为(pytorch)开头
版本检查
在命令提示符窗口输入命令:nvidia-smi
显示NVIDIA-SMI 461.40 CUDA Version 11.2
安装PyTorch
PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/
进行如下选择并复制命令(该PyTorch版本为1.11)
在Anaconda Prompt的pytorch环境中粘贴执行此命令开始安装
安装完成后可通过pip list命令来查看
检查是否成功安装