1:先安装anaconda,直接从官网中搜索安装anaconda3即可;
2:安装nvidia驱动程序:
找准显卡的版本,安装对应版本的nvidia驱动程序
3:安装cuda+torch
在conda prompt中先创建一个单独的pytorch环境
conda creat -n pytorch python=3.8
激活进入该环境,在该环境中进行安装torch和cuda,安装的方法如下 ;
next
在该环境中安装torch+cuda ;(先进入torch官网,如图选取自己nvidia支持的cuda版本,并且采用conda形式安装,直接复制最后一行命令,然后在prompt中执行此条命令语句。估计花费不少时间,之后cuda 和 torch就自动安装在名为pytorch的环境下,用conda list 命令语句可以查看安装后的信息)
next:
验证torch和cuda是否安装成功:
在名为pytroch的环境下,进入python环境,然后在python环境下运行以下语句
import torch
torch.cuda.is_available()
如果输出true,则代表大功告成
如下图所示:
final:
安装pycharm ;
直接从官网下载安装,无需配环境