TensorFlow,PyTorch均已用anaconda安装到虚拟环境中,地址为D:\Anaconda\envs\各个库的名字\Lib\site-packages
Gym在vscode终端中通过pip install 安装(除了box2d都已安装成功),但无法调用。
vscode所使用的库的地址:C:\Users\24061\AppData\Local\Programs\Python\Python310\Lib\site-packages
win11;vscode;显卡为RTX3060,驱动最新;PyTorch下载的是GPU版本;Gym采用pip install 安装后box2d安装失败(pip install --upgrade gym[box2d]无法解决)。
1.在vscode设置中设置conda path,为D:\Anaconda,只有TensorFlow能正常使用,另两个库都无法使用。
2.在vscode设置中设置Default Interpreter Path,无用。
运行成功
例子程序:
import tensorflow
import torch
import gym
#测试tensorflow
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
#测试显卡是否可用
print(torch.cuda.is_available())
#测试gym
env = gym.make('CartPole-v0')
env.reset()
for _ in range(1000):
env.render()
env.step(env.action_space.sample()) # take a random action
env.close()
#1.输出:Hello, TensorFlow!
#2.输出:True
#3.输出:小车倒立摆模拟
你可以去虚拟环境库中 地址为D:\Anaconda\envs\各个库的名字\Lib\site-packages这里面
把需要的复制到vscode当前用的site-packages下,
也就是这里C:\Users\24061\AppData\Local\Programs\Python\Python310\Lib\site-packages