1.安装anaconda
- 创建环境 pytorch是环境名字,后面是py的版本,可以自己更改。
-
conda create -n pytorch python=3.9.7
2.安装pytorch
这一步根据自己的GPU选择合适的版本,没有GPU的可以选择CPU的版本。
PyTorch Build : 默认选择Stable即可。
Your OS :选择你对应的操作系统。
Package :这里建议选择pip模式(因为其他模式可能会有点问题)。
Language :看你自己的情况。
CUDA的版本看自己电脑支持的版本。没有GPU的可以直接选择后面的CPU版本。
一些问题:
- 如何查看自己的电脑支持的CUDA版本
step1:右键点击NVIDIA控制面板 ,
step2: