踩到的坑
1.安装pytorch时用的不是官方的源,换成了清华源,然后运行的时候无法支持cuda;换成官网的可以。
2.numpy版本为2.0,torch有些功能不兼容,可以降低版本。我的python为3.9,numpy版本改为了1.26,实测可用。
pip install numpy==1.26
Python 2.7的最后一个版本 | NumPy 1.16.5 |
Python 3.5的最后一个版本 | NumPy 1.19.5 |
Python 3.6及以上版本 | NumPy 1.20.3 |
安装步骤
1.安装anaconda
创建conda虚拟环境
2.查看可安装的cuda版本并安装(注意GPU的算力,我的算力是457.63,只能安装11.1以下的CUDA版本)
3.去官网下载符合cuda版本的pytorch版本
安装完毕。
在pycharm中调用torch