本文提供了详细的步骤指导如何为PyTorch创建一个合适的运行环境。首先通过conda命令创建指定Python版本的虚拟环境，并激活该环境。接着确认CUDA版本，以便在PyTorch官网找到匹配的安装指令。最后通过Python脚本验证PyTorch是否正确安装。
第一步
conda create -n 你想给环境起的名字 python=版本号3.7or3.8(等号中间不要有空格)
第二步
conda activate 你的名字
第三步
看cuda的版本，命令是：nvidia-smi
查看静态版本 nvcc --version
第四步：
去pytorch上找对应cuda版本的安装命令,在红字的previous versions里面找
第五步
把命令复制过来执行
第六步
依次输入
python
import torch
torch.cuda.is_available()
如果返回True就是成功
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
