由于学业上会用到PyTorch,所以今天花了一些时间来安装,实际也踩了一些坑。
我原本已经下载好了anaconda,实际下载也很简单,故不过多赘述。
我自己的python版本是3.9
1.进入anaconda prompt
输入
nvidia-smi
就会呈现出你cuda的版本号,本人是11.7
这里要注意,我原本的版本号是11.0,但是pytorch官网上适合我的版本只有10.2(如下图),(须知 所选版本必须要低于自己的版本)所以我建议可以去官网下载一下最新版的cuda,我下载了11.7的,因此在pytorch官网上可以选择cuda11.6
2.创建虚拟环境
在anaconda prompt上输入
conda create -n PyTorch python=3.9
PyTorch是我定义的环境名称
我的python版本是3.9
故遵从格式:
conda create -n xxxxx&#