南昌微信网站建设,长沙诚信做网站,龙斗seo博客,百度关键词【超级简单】3步 安装conda pytorch gpu版本 1. 创建虚拟环境至于python 3.X和cuda对应 #xff1f; 2. 下载cuda 驱动#xff0c;cuda toolkit (可选)3. 虚拟环境中输入以下指令至于 怎么查看自己电脑对应的cuda版本号 #xff1f; 4.测试是否安装成功有疑惑可以再看看这篇… 【超级简单】3步 安装conda pytorch gpu版本 1. 创建虚拟环境至于python 3.X和cuda对应 2. 下载cuda 驱动cuda toolkit (可选)3. 虚拟环境中输入以下指令至于 怎么查看自己电脑对应的cuda版本号 4.测试是否安装成功有疑惑可以再看看这篇文章写的。 下载anaconda能让你高效管理不同版本运行环境。 conda pytorch gpu 适用于linux 和window版本 建议前提确保自己的显卡能用哈。
以下是在window10的示范。
1. 创建虚拟环境
conda create -n your_env_name python3.X conda avtivate your_env_name
至于python 3.X和cuda对应
参考Tensorflow 的资料 https://www.tensorflow.org/install/source_windows?hlzh-cn
2. 下载cuda 驱动cuda toolkit (可选)
https://developer.nvidia.com/cuda-toolkit-archive
下载成功的话可以 在cmd 输入 nvcc -V, 显示版本号。
显示 not found 则说明没转成功。
3. 虚拟环境中输入以下指令
conda install pytorch torchvision torchaudio cudatoolkitX.X -c pytorchX.X替换成你的版本号
至于 怎么查看自己电脑对应的cuda版本号
鼠标右键——NVDIA控制面板——帮助——系统信息——组件product name 那一列就是nvdia对应版本。
4.测试是否安装成功
torch.cuda.is_avaliable()
# True, gpu is ok
# false, try to fix it by rebooting up. if not, there are some pro. with your settings, like conda environment doesnt correct, cuda Ver. is not right.
有疑惑可以再看看这篇文章写的。
笔墨多一点。 具体可参考https://blog.csdn.net/qq_36851515/article/details/107608967
